Fixed nccl_op.cc

8ab2b5ce · Kavya Srinet · 7343b20d · 8ab2b5ce
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/nccl_op.cc paddle/fluid/operators/nccl_op.cc +1 -1

未找到文件。
--- a/paddle/fluid/operators/nccl_op.cc
+++ b/paddle/fluid/operators/nccl_op.cc
@@ -55,7 +55,7 @@ class NCCLInitOpMaker : public framework::OpProtoAndCheckerMaker {
    AddAttr<int>("dtype",
                 "(int, default 5 (FP32)) "
                 "Output data type")
-        .SetDefault(framework::proto::DataType::FP32);
+        .SetDefault(framework::proto::VarType::FP32);
    AddComment(R"DOC(
 NCCLInit Operator.