fix devices registry name error for Cambricon and Hygon this two type

Signed-off-by: rongfu.leng <[email protected]>
Project-HAMi · Mar 15, 2024 · 73a0bdf · 73a0bdf
1 parent 58275e5
commit 73a0bdf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/device/devices.go b/pkg/device/devices.go
@@ -45,10 +45,10 @@ func GetDevices() map[string]Devices {
 
 func init() {
 	devices = make(map[string]Devices)
-	devices["Cambricon"] = cambricon.InitMLUDevice()
-	devices["NVIDIA"] = nvidia.InitNvidiaDevice()
-	devices["Hygon"] = hygon.InitDCUDevice()
-	devices["Iluvatar"] = iluvatar.InitIluvatarDevice()
+	devices[cambricon.CambriconMLUDevice] = cambricon.InitMLUDevice()
+	devices[nvidia.NvidiaGPUDevice] = nvidia.InitNvidiaDevice()
+	devices[hygon.HygonDCUDevice] = hygon.InitDCUDevice()
+	devices[iluvatar.IluvatarGPUDevice] = iluvatar.InitIluvatarDevice()
 	DevicesToHandle = []string{}
 	DevicesToHandle = append(DevicesToHandle, nvidia.NvidiaGPUCommonWord)
 	DevicesToHandle = append(DevicesToHandle, cambricon.CambriconMLUCommonWord)