Fix get device context error (PaddlePaddle#34)

Co-authored-by: jianghaicheng <haichengj@graphcore.ai>
graphcore · Aug 9, 2021 · 5fc456d · 5fc456d
1 parent 3af0a36
commit 5fc456d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/operators/ipu_runtime_op.cc b/paddle/fluid/operators/ipu_runtime_op.cc
@@ -46,6 +46,6 @@ Run graph by PopART runtime.
 }  // namespace paddle
 
 namespace ops = paddle::operators;
-REGISTER_OPERATOR(ipu_runtime, ops::IpuRuntimeOp,
-                  ops::IpuRuntimeOpMaker);
+REGISTER_OPERATOR(ipu_runtime, ops::IpuRuntimeOp, ops::IpuRuntimeOpMaker);
 REGISTER_OP_CPU_KERNEL(ipu_runtime, ops::IpuRuntimeKernel<float>)
+REGISTER_OP_IPU_KERNEL(ipu_runtime, ops::IpuRuntimeKernel<float>)