@daocloud-proto/hydra 0.18.0-dev-5 → 0.18.0-dev-8

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

@@ -42,6 +42,20 @@ export type ListGPUTypesResponse = {
 export type GPUType = {
   gpuName?: string
   gpuType?: string
+  displayName?: string
+  provider?: string
+  resources?: GPUResourceSchema[]
+}
+
+export type GPUResourceSchema = {
+  key?: string
+  displayName?: string
+  displayNameZh?: string
+  semanticType?: string
+  minimum?: string
+  maximum?: string
+  description?: string
+  descriptionZh?: string
 }
 
 export class Management {

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

@@ -204,15 +204,21 @@ export type ModelServingSpec = {
   deployTemplates?: DeployTemplate[]
 }
 
-export type DeployTemplateResourcesRequirements = {
+
+/* hydra modified */ export type BaseDeployTemplateResourcesRequirements = {
   cpu?: number
   memory?: number
   gpuType?: string
   gpuCount?: number
   perGpuMemory?: number
   gpuCores?: number
+  gpuResourceValues?: {[key: string]: string}
 }
 
+export type DeployTemplateResourcesRequirements = BaseDeployTemplateResourcesRequirements
+  & OneOf<{ shmSize: number }>
+  & OneOf<{ enableShm: boolean }>
+
 export type DeployTemplate = {
   resourcesRequirements?: DeployTemplateResourcesRequirements
   runtime?: DeployTemplateInferenceRuntime

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

@@ -147,10 +147,13 @@ export type ConfigMapMountConfig = {
   gpuType?: string
   gpuCount?: number
   perGpuMemory?: number
+  gpuResourceValues?: {[key: string]: string}
 }
 
 export type ResourceConfig = BaseResourceConfig
   & OneOf<{ gpuCores: number }>
+  & OneOf<{ shmSize: number }>
+  & OneOf<{ enableShm: boolean }>
 
 export type EnvVar = {
   name?: string

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.18.0-dev-5",
+  "version": "v0.18.0-dev-8",
   "description": "",
   "author": "",
   "license": "ISC"