forked from huawei/mindspore2022
45 lines
1.4 KiB
Protocol Buffer
45 lines
1.4 KiB
Protocol Buffer
/**
|
||
* Copyright 2021 Huawei Technologies Co., Ltd
|
||
*
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
// The anf_ir.proto is `proto2` syntax, so this file use `proto2`, else will occur a compile error.
|
||
syntax = "proto2";
|
||
|
||
package mindspore.irpb;
|
||
option cc_enable_arenas = true;
|
||
|
||
import "anf_ir.proto";
|
||
|
||
|
||
message ProfilingParallel {
|
||
optional string version = 1;
|
||
optional Config config = 2;
|
||
optional GraphProto graph = 3;
|
||
}
|
||
|
||
|
||
message Config {
|
||
optional uint32 rank_id = 1;
|
||
|
||
// The stage of current device
|
||
optional uint32 stage_id = 2;
|
||
|
||
// optional： stand_alone/data_parallel/semi_auto_parallel/auto_parallel/hybrid_parallel
|
||
optional string parallel_type = 3;
|
||
|
||
// The global stages and devices info, ex. 2 stages: [[0,1,2,3], [4,5,6,7]]
|
||
// If the training mode is not pipeline parallel training, [[0,1,2,3,4,5,6,7,8]]
|
||
repeated TensorShapeProto stage_devices = 4;
|
||
} |