fix save_vars bugs (#40062)

eaacf8bf · YuanRisheng · GitHub · e7aea650 · eaacf8bf
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

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

未找到文件。
--- a/paddle/fluid/operators/save_combine_op.cc
+++ b/paddle/fluid/operators/save_combine_op.cc
@@ -38,7 +38,8 @@ class SaveCombineOp : public framework::OperatorWithKernel {
  framework::OpKernelType GetKernelTypeForVar(
      const std::string& var_name, const Tensor& tensor,
      const framework::OpKernelType& expected_kernel_type) const override {
-    return expected_kernel_type;
+    return framework::OpKernelType(expected_kernel_type.data_type_,
+                                   tensor.place());
  }
 };