Fix

PaddlePaddle · Jan 10, 2025 · 7ec5eba · 7ec5eba
1 parent 9d3cb3d
commit 7ec5eba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/inference/tensorrt/plugin/c_allreduce_op_plugin.cu b/paddle/fluid/inference/tensorrt/plugin/c_allreduce_op_plugin.cu
@@ -197,13 +197,13 @@ int CAllReducePluginDynamic::enqueue(
                         "NCCLCommContext is nullptr, collective op should "
                         "has ring_id attr."));
   auto stream2 = comm_ctx->GetStream();
-  // ncclRedOp_t nccl_red_type = ncclSum;
+  ncclRedOp_t nccl_red_type2 = ncclSum;
   // comm_ctx->AllReduce(&inputs[0], inputs[0], nccl_red_type, stream);
   phi::dynload::ncclAllReduce(sendbuff,
                               recvbuff,
                               numel,
                               dtype,
-                              nccl_red_type,
+                              nccl_red_type2,
                               comm_ctx->GetNcclComm(),
                               stream2);
   VLOG(3) << "new NCCLCommContext has ring_id_ " << ring_id_;