| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | device_matrix.cc | 111 DeviceMatrix dev_matrix(rank_, dev_list_, dev_shape_); in GetDevicesAlongMultiDim() local
118 DeviceMatrix dev_matrix(rank, dev_list_, dev_shape_); in GetDevicesAlongMultiDim() local
|
| D | came_parallel_handler.h | 63 DeviceMatrix dev_matrix; variable
|
| D | came_parallel_handler.cc | 250 DeviceMatrix dev_matrix(rank, full_rank_list, tensor_layout->device_arrangement().array()); in GetDimRankList() local
|
| D | parameter_manager.cc | 227 static RankList GetDevListByTensorMapValue(DeviceMatrix dev_matrix, int64_t tensor_map_value, size_… in GetDevListByTensorMapValue()
441 DeviceMatrix dev_matrix(rank, rank_list, dev_mat_shape); in IsFullySplitParameter() local
1054 DeviceMatrix dev_matrix(rank, stage_device_list, dev_shape); in GetRankListByLayout() local
|
| D | step_parallel_utils.cc | 2867 DeviceMatrix dev_matrix(rank, stage_dev_list, layout->device_arrangement().array()); in mirror_group_list() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | strategy_checkpoint_info.cc | 44 auto dev_matrix = layout_j.value().at("dev_matrix").get<std::vector<int64_t>>(); in FromJson() local
152 straspb::DevMatrix *dev_matrix = parallel_layouts->add_dev_matrix(); in to_protobuf() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/ |
| D | node_strategy.proto | 52 repeated DevMatrix dev_matrix = 1; field
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | flash_attention_score_info.cc | 92 int64_t GetSplitNumByMapId(const Shape &dev_matrix, int64_t map_id) { in GetSplitNumByMapId()
104 int64_t GetSplitNumByTensorMap(const Shape &dev_matrix, const Shape &tensor_map) { in GetSplitNumByTensorMap()
184 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in GetSPRankList() local
926 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in GetSplitIdAndRank() local
|
| D | fused_infer_attention_score_info.cc | 349 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in GetSplitIdAndRank() local
644 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in ComputeReplaceGraphForSplitKVSeq() local
|
| D | gatherd_info.cc | 266 DeviceMatrix dev_matrix(rank, g_device_manager->GetDeviceListInThisStage(), dev_matrix_shape_); in InferGroup() local
|
| D | crop_and_resize_info.cc | 122 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferGroup() local
|
| D | roi_align_info.cc | 131 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferGroup() local
|
| D | resizebilinear_v2_info.cc | 225 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in InferRankBias() local
|
| D | prompt_flash_attention_info.cc | 201 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in GetSplitIdAndRank() local
|
| D | operator_info.cc | 1046 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupByTensorMap() local
1086 …DeviceMatrix dev_matrix(rank, stage_device_list_, tensor_layout->device_arrangement_origin().array… in CreateGroupForOptShard() local
1243 DeviceMatrix dev_matrix(rank, stage_device_list_, dev_matrix_shape_); in CreateGroupByDim() local
1248 Status OperatorInfo::CreateGroupByDimWithDevMatrix(DeviceMatrix *dev_matrix, size_t axis, std::vect… in CreateGroupByDimWithDevMatrix()
|
| D | max_avg_pool_3d_info.cc | 409 DeviceMatrix dev_matrix(rank_id, stage_device_list_, dev_matrix_shape_); in GetAdjacentRankIdsAndBiases() local
|
| D | reshape_info.cc | 535 Arrangement dev_matrix; in InferTensorLayout() local
|
| D | conv2d_info.cc | 456 DeviceMatrix dev_matrix(rank_id, stage_device_list_, dev_matrix_shape_); in GetAdjacentRankIdsAndBiases() local
|
| D | gather_info.cc | 883 DeviceMatrix dev_matrix(rank, g_device_manager->GetDeviceListInThisStage(), dev_matrix_shape_); in InferGroup() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _tensor.py | 389 def _construct_tensor_layout_for_opt_shard(dev_matrix, tensor_map, opt_shard_step, opt_shard_size, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_layout.cc | 645 …DeviceMatrix dev_matrix(rank, g_device_manager->GetDeviceListInThisStage(), device_arrangement_ori… in InferRepeatedGroup() local
|
| D | construct_operator.cc | 359 DeviceMatrix dev_matrix(rank, dev_list_, dev_matrix_shape_); in CreateGroupByDim() local
|