fix: check driver command for adding worker

This commit is contained in:
jialin
2025-10-20 19:11:02 +08:00
parent 9b843151fe
commit 3a7a4ccf30
12 changed files with 162 additions and 61 deletions
+3 -9
View File
@@ -91,20 +91,14 @@ export const gpuColorMap: Record<string, string> = {
rocm: 'volcano',
dtk: 'magenta',
musa: 'cyan',
corex: 'purple'
corex: 'purple',
maca: 'geekblue'
};
export const getGpuColor = (gpuType: string) => {
if (!gpuType) return 'default';
if (gpuType.includes('cann')) return gpuColorMap['cann'];
if (gpuType.includes('cuda')) return gpuColorMap['cuda'];
if (gpuType.includes('rocm')) return gpuColorMap['rocm'];
if (gpuType.includes('dtk')) return gpuColorMap['dtk'];
if (gpuType.includes('musa')) return gpuColorMap['musa'];
if (gpuType.includes('corex')) return gpuColorMap['corex'];
return 'default';
return gpuColorMap[gpuType] || 'default';
};
export const customIcons = [