fix linspace (#41404)

PaddlePaddle · Apr 5, 2022 · 84e8ae7 · 84e8ae7
1 parent 870402f
commit 84e8ae7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/fluid/operators/linspace_op.cc b/paddle/fluid/operators/linspace_op.cc
@@ -38,7 +38,8 @@ class LinspaceOp : public framework::OperatorWithKernel {
   framework::OpKernelType GetKernelTypeForVar(
       const std::string &var_name, const framework::Tensor &tensor,
       const framework::OpKernelType &expected_kernel_type) const override {
-    return expected_kernel_type;
+    return framework::OpKernelType(expected_kernel_type.data_type_,
+                                   tensor.place(), tensor.layout());
   }
 };