fix

PaddlePaddle · Mar 9, 2025 · ae6db60 · ae6db60
1 parent bea2fa4
commit ae6db60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/pybind/process_group_utils.h b/paddle/fluid/pybind/process_group_utils.h
@@ -113,12 +113,12 @@ struct SplitDenseTensor<phi::CustomContext, T> {
       std::vector<int64_t> new_dims = {tensor->numel()};
       if (tensor_dims.size() != new_dims.size()) {
         // flatten
-        tensor->Resize();
+        tensor->Resize(common::new_dims(new_dims));
       }
     }
     (*kernel_fn)(
         context, in_flatten, phi::IntArray(sections), phi::Scalar(0), *out);
-    if (in_dims != flattened_in_dims) {
+    if (in_dims.size() != flattened_in_dims.size()) {
       in.Resize(in_dims);
     }
     for (int64_t i = 0; i < out->size(); ++i) {