[ascend]adapt ray (#225)

Author: JackWeiw

dlinfer/graph/dicp/vendor/AtbGraph/codegen/atb_op.py

@@ -60,6 +60,7 @@ def LinearAllReduce(name, x, weight, bias, group):
         param.hasResidual = False
         param.parallelType = infer_param.ParallelType.LINEAR_ALL_REDUCE
         param.commMode = infer_param.CommMode.COMM_MULTI_PROCESS
+        param.commDomain = group if group is not None else ""
 
         # LCCL has issues with multiple communication domains. By default,
         # in single-machine multi-card scenarios, LCCL is enabled.
@@ -75,7 +76,6 @@ def LinearAllReduce(name, x, weight, bias, group):
             param.backend = "lccl"
         else:
             param.backend = "hccl"
-            param.commDomain = group if group is not None else ""
             if rank_table_file is not None:
                 param.rankTableFile = rank_table_file
 
@@ -95,6 +95,7 @@ def AllReduce(name, x, reduce_type, group):
         param.rankRoot = 0
         param.allReduceType = reduce_type
         param.commMode = infer_param.CommMode.COMM_MULTI_PROCESS
+        param.commDomain = group if group is not None else ""
 
         # LCCL has issues with multiple communication domains. By default,
         # in single-machine multi-card scenarios, LCCL is enabled.
@@ -110,7 +111,6 @@ def AllReduce(name, x, reduce_type, group):
             param.backend = "lccl"
         else:
             param.backend = "hccl"
-            param.commDomain = group if group is not None else ""
             if rank_table_file is not None:
                 param.rankTableFile = rank_table_file