fix(ascend): use CommInitAll for single-node TP

Author: baominghelly

csrc/engine/distributed/communication_group.cpp

@@ -176,7 +176,10 @@ CommunicationGroup::CommunicationGroup(const DistConfig &dist_config,
     }
 
     if (tp_size > 1) {
-        if (device_type_ == infinicore::Device::Type::kCambricon) {
+        const bool use_single_node_init_all =
+            device_type_ == infinicore::Device::Type::kCambricon
+            || device_type_ == infinicore::Device::Type::kAscend;
+        if (use_single_node_init_all) {
             checkInfiniccl(
                 "infinicclCommInitAll",
                 infinicclCommInitAll(

test/static/test_infinicore_runtime_contracts.py

@@ -56,6 +56,24 @@ def test_eager_graph_tensors_alias_while_compiled_outputs_force_blobs(
                     source.count("GraphTensor::SnapshotPolicy::kBlob"), count
                 )
 
+    def test_ascend_single_node_tp_uses_comm_init_all(self) -> None:
+        source = read_source(
+            "csrc/engine/distributed/communication_group.cpp"
+        )
+        constructor = function_body(
+            source,
+            "CommunicationGroup::CommunicationGroup(const DistConfig &dist_config,",
+        )
+
+        self.assertIn("const bool use_single_node_init_all", constructor)
+        self.assertIn(
+            "device_type_ == infinicore::Device::Type::kAscend",
+            constructor,
+        )
+        self.assertIn("infinicclCommInitAll(", constructor)
+        self.assertIn("communicators_", constructor)
+        self.assertNotIn("infinicclCommInitRank", constructor)
+
     def test_standard_mlp_consumes_packed_gate_up_without_repacking(self) -> None:
         consumers = (
             (