@daocloud-proto/baize 0.112.0 → 0.114.0

package/common/k8s.pb.ts

@@ -56,6 +56,7 @@ export type PodConfig = {
   queue?: string
   tolerationSeconds?: string
   resourceAffinities?: ResourceAffinity[]
+  shmSize?: number
 }
 
 export type Affinity = {

package/management-api/job/v1alpha1/job.pb.ts

@@ -82,6 +82,22 @@ export type ListJobsResponse = {
   page?: BaizeCommonCommon.Pagination
 }
 
+export type ListRdmasRequest = {
+  workspace?: number
+  cluster?: string
+  namespace?: string
+  page?: BaizeCommonCommon.Pagination
+}
+
+export type Rdma = {
+  name?: string
+}
+
+export type ListRdmasResponse = {
+  items?: Rdma[]
+  page?: BaizeCommonCommon.Pagination
+}
+
 
 /* baize modified */ export type BaseSingleJobRequest = {
   workspace?: number
@@ -111,6 +127,7 @@ export type CreateJobRequest = {
 
 
 /* baize modified */ export type BaseTrainingConfig = {
+  rdmaEnabled?: boolean
 }
 
 export type TrainingConfig = BaseTrainingConfig
@@ -177,4 +194,7 @@ export class JobsManagement {
   static ListSchedulers(req: CreateJobRequest, initReq?: fm.InitReq): Promise<JobSchedulersResponse> {
     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)})
   }
+  static ListRdmas(req: ListRdmasRequest, initReq?: fm.InitReq): Promise<ListRdmasResponse> {
+    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"})
+  }
 }

package/management-api/notebook/v1alpha1/notebook.pb.ts

@@ -80,6 +80,7 @@ export type NotebookConfig = {
   affinity?: BaizeCommonK8s.Affinity
   tolerationSeconds?: string
   imageConfig?: BaizeManagement_apiImageV1alpha1Image.ImageConfig
+  shmSize?: number
 }
 
 export type NotebookStatus = {

package/management-api/queue/v1alpha1/queue.pb.ts

@@ -83,6 +83,7 @@ export type Queue = {
   workspaceAlias?: string
   preemptionWithinClusterQueue?: QueuePreemptionWithinClusterQueuePolicy
   flavors?: FlavorQuota[]
+  tasEnabled?: boolean
 }
 
 export type ListQueueRequest = {
@@ -201,6 +202,7 @@ export type ListQueueWorkloadResponse = {
 
 export type ResourceFlavor = {
   name?: string
+  topologyName?: string
 }
 
 export type ListKueueResourceFlavorsRequest = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/baize",
-  "version": "v0.112.0",
+  "version": "v0.114.0",
   "description": "",
   "author": "",
   "license": "ISC"