add support for cuda

51b30932 · ashburnlee · f9f3bc21 · 51b30932
显示空白变更内容
内联并排

Showing with 1 addition and 2 deletion

paddle/fluid/operators/unique_op.cc paddle/fluid/operators/unique_op.cc +1 -2

未找到文件。
--- a/paddle/fluid/operators/unique_op.cc
+++ b/paddle/fluid/operators/unique_op.cc
@@ -88,8 +88,7 @@ class UniqueOp : public framework::OperatorWithKernel {
  framework::OpKernelType GetExpectedKernelType(
      const framework::ExecutionContext& ctx) const override {
    return framework::OpKernelType(
-        OperatorWithKernel::IndicateVarDataType(ctx, "X"),
+        OperatorWithKernel::IndicateVarDataType(ctx, "X"), ctx.GetPlace());
-        platform::CPUPlace());
  }
 };