Searched refs:data_size_list (Results 1 – 1 of 1) sorted by relevance
104 def _set_fusion_strategy_by_size(data_size_list, group="hccl_world_group"): argument142 if isinstance(data_size_list, list):143 len_data_size = len(data_size_list)148 for data_size in data_size_list:152 c_array_size_list = _c_array(ctypes.c_float, data_size_list)153 c_size_num = ctypes.c_uint(len(data_size_list))