style: backend list styles

2025-10-13 14:40:26 +08:00
parent 0dfe77df37
commit eb509166ee
18 changed files with 209 additions and 360 deletions
@@ -101,6 +101,19 @@ export const gpuColorMap: Record<string, string> = {
  corex: 'purple'
 };

+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';
+};
+
 export const customIcons = ['∑', '∏', '∫', '∂', '∞', 'θ', '∆', '∇'];

 export const backendFields = [