@daocloud-proto/baize 0.112.0 → 0.113.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
|
@@ -82,6 +82,22 @@ export type ListJobsResponse = {
|
|
|
82
82
|
page?: BaizeCommonCommon.Pagination
|
|
83
83
|
}
|
|
84
84
|
|
|
85
|
+
export type ListRdmasRequest = {
|
|
86
|
+
workspace?: number
|
|
87
|
+
cluster?: string
|
|
88
|
+
namespace?: string
|
|
89
|
+
page?: BaizeCommonCommon.Pagination
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
export type Rdma = {
|
|
93
|
+
name?: string
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
export type ListRdmasResponse = {
|
|
97
|
+
items?: Rdma[]
|
|
98
|
+
page?: BaizeCommonCommon.Pagination
|
|
99
|
+
}
|
|
100
|
+
|
|
85
101
|
|
|
86
102
|
/* baize modified */ export type BaseSingleJobRequest = {
|
|
87
103
|
workspace?: number
|
|
@@ -111,6 +127,7 @@ export type CreateJobRequest = {
|
|
|
111
127
|
|
|
112
128
|
|
|
113
129
|
/* baize modified */ export type BaseTrainingConfig = {
|
|
130
|
+
rdmaEnabled?: boolean
|
|
114
131
|
}
|
|
115
132
|
|
|
116
133
|
export type TrainingConfig = BaseTrainingConfig
|
|
@@ -177,4 +194,7 @@ export class JobsManagement {
|
|
|
177
194
|
static ListSchedulers(req: CreateJobRequest, initReq?: fm.InitReq): Promise<JobSchedulersResponse> {
|
|
178
195
|
return fm.fetchReq<CreateJobRequest, JobSchedulersResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/jobs/${req["name"]}/schedulers`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
|
|
179
196
|
}
|
|
197
|
+
static ListRdmas(req: ListRdmasRequest, initReq?: fm.InitReq): Promise<ListRdmasResponse> {
|
|
198
|
+
return fm.fetchReq<ListRdmasRequest, ListRdmasResponse>(`/apis/baize.io/v1alpha1/workspaces/${req["workspace"]}/clusters/${req["cluster"]}/namespaces/${req["namespace"]}/rdmas?${fm.renderURLSearchParams(req, ["workspace", "cluster", "namespace"])}`, {...initReq, method: "GET"})
|
|
199
|
+
}
|
|
180
200
|
}
|