From 51b3093230b3e2db09cf2fed5c42fc90190212d9 Mon Sep 17 00:00:00 2001
From: ashburnlee <1578034415@qq.com>
Date: Thu, 24 Sep 2020 04:36:57 +0000
Subject: [PATCH] add support for cuda

---
 paddle/fluid/operators/unique_op.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/unique_op.cc b/paddle/fluid/operators/unique_op.cc
index 745102dd28d..45ab2dd22cc 100644
--- 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"),
-        platform::CPUPlace());
+        OperatorWithKernel::IndicateVarDataType(ctx, "X"), ctx.GetPlace());
   }
 };
 
-- 
GitLab