diff --git a/src/assets/logo/t-head.png b/src/assets/logo/t-head.png new file mode 100644 index 00000000..60fb11fe Binary files /dev/null and b/src/assets/logo/t-head.png differ diff --git a/src/pages/backends/config/index.ts b/src/pages/backends/config/index.ts index 2781bb1e..33e06a2d 100644 --- a/src/pages/backends/config/index.ts +++ b/src/pages/backends/config/index.ts @@ -194,6 +194,12 @@ export const frameworks = [ tips: ManufacturerMap[GPUDriverMap.CAMBRICON], tipLocale: true }, + { + label: 'HGGC', + value: GPUDriverMap.THEAD, + tips: ManufacturerMap[GPUDriverMap.THEAD], + tipLocale: true + }, { label: 'CPU', value: 'cpu' diff --git a/src/pages/cluster-management/components/support-gpus.tsx b/src/pages/cluster-management/components/support-gpus.tsx index 43098962..1fbdcd48 100644 --- a/src/pages/cluster-management/components/support-gpus.tsx +++ b/src/pages/cluster-management/components/support-gpus.tsx @@ -5,6 +5,7 @@ import iluvatarWEBP from '@/assets/logo/Iluvatar.png'; import metaxLogo from '@/assets/logo/metax.png'; import mooreLogo from '@/assets/logo/moore-logo.png'; import nvidiaLogo from '@/assets/logo/nvidia.png'; +import theadLogo from '@/assets/logo/t-head.png'; import IconFont from '@/components/icon-font'; import useUserSettings from '@/hooks/use-user-settings'; import { @@ -189,12 +190,7 @@ const SupportedHardware: React.FC = ({ locale: false, link: 'https://docs.gpustack.ai/latest/installation/requirements/#thead-gpu', notes: AddWorkerDockerNotes[GPUDriverMap.THEAD], - icon: ( - - ) + icon: } ]; diff --git a/src/pages/resources/config/gpu-driver.ts b/src/pages/resources/config/gpu-driver.ts index e4ab3118..88219dfa 100644 --- a/src/pages/resources/config/gpu-driver.ts +++ b/src/pages/resources/config/gpu-driver.ts @@ -77,7 +77,7 @@ export const GPUsConfigs: Record< [GPUDriverMap.THEAD]: { label: 'T-Head', value: GPUDriverMap.THEAD, - runtime: 'thead', // TODO: confirm runtime name + runtime: '', // TODO: confirm runtime name driver: 'ppu-smi' } }; @@ -250,7 +250,6 @@ const registerTHeadWorker = (params: AddWorkerCommandParams) => { return `${commonArgs} --volume /usr/local/PPU_SDK:/usr/local/PPU_SDK:ro \\ --volume /var/run/cdi:/var/run/cdi \\ - --runtime ${config.runtime} \\ ${imageArgs} ${setWorkerIPArg(params)}`; };