Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions protos/config.proto
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,8 @@ message Partition {

// 适配器配置的,每个Pod加速卡数量限额
optional uint32 max_accelerators_per_pod = 12;
// cpu_model: 具体的 cpu 型号,如 "Intel(R) Xeon(R) Platinum 81633090" 等
optional string cpu_model = 13;
}

message GetClusterConfigResponse {
Expand Down
7 changes: 7 additions & 0 deletions protos/job.proto
Original file line number Diff line number Diff line change
Expand Up @@ -235,6 +235,11 @@ message EnvVariable {
string value = 2;
}

message PrivateImageRepository {
string user_name = 1;
string password = 2;
}

message SubmitJobRequest {
string user_id = 1;
string job_name = 2;
Expand Down Expand Up @@ -275,6 +280,8 @@ message SubmitJobRequest {
repeated EnvVariable env_variables = 18;
// 训练中TensorBoard的数据源路径
optional string tensor_board_data_path = 19;
// 镜像私仓的用户名和密码
optional PrivateImageRepository private_image_repository = 20;
}

message SubmitJobResponse {
Expand Down