Merge pull request #31 from cybergeek2077/main

Register device memory using the GPU memory factor
Project-HAMi · Oct 15, 2024 · 86d7a26 · 86d7a26
2 parents 506be8b + 9dd6650
commit 86d7a26
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/plugin/vgpu/register.go b/pkg/plugin/vgpu/register.go
@@ -62,7 +62,8 @@ func (r *DeviceRegister) apiDevices() *[]*util.DeviceInfo {
 		} else {
 			klog.V(3).Infoln("nvml registered device id=", dev.ID, "memory=", *ndev.Memory, "type=", *ndev.Model)
 		}
-		registeredmem := int32(*ndev.Memory)
+		registeredmem := int32(*ndev.Memory) / int32(config.GPUMemoryFactor)
+		klog.V(3).Infoln("GPUMemoryFactor=", config.GPUMemoryFactor, "registeredmem=", registeredmem)
 		res = append(res, &util.DeviceInfo{
 			Id:     dev.ID,
 			Count:  int32(config.DeviceSplitCount),