mindspore2022/mindspore/ccsrc/utils/profiling_parallel.proto

/**
 * 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;
}