SKS-2160: Fix 'where' condition when getting GPU device allocation de…

…tails (#157)
smartxworks · Dec 1, 2023 · 2fe52eb · 2fe52eb
1 parent 7740e82
commit 2fe52eb
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/pkg/service/vm.go b/pkg/service/vm.go
@@ -970,21 +970,19 @@ func (svr *TowerVMService) GetGPUDevicesAllocationInfoByHostIDs(hostIDs []string
 		return NewGPUVMInfos(), nil
 	}
 
-	where := &models.GpuDeviceWhereInput{
-		UserUsage: models.NewGpuDeviceUsage(gpuDeviceUsage),
-		Host: &models.HostWhereInput{
-			IDIn: hostIDs,
+	getDetailVMInfoByGpuDevicesParams := clientgpu.NewGetDetailVMInfoByGpuDevicesParams()
+	getDetailVMInfoByGpuDevicesParams.RequestBody = &models.GetGpuDevicesRequestBody{
+		Where: &models.GpuDeviceWhereInput{
+			UserUsage: models.NewGpuDeviceUsage(gpuDeviceUsage),
+			Host: &models.HostWhereInput{
+				IDIn: hostIDs,
+			},
 		},
 	}
 
 	// Filter GPU devices whose vGPU has been fully used
 	if gpuDeviceUsage == models.GpuDeviceUsageVGPU {
-		where.AvailableVgpusNumGt = TowerInt32(0)
-	}
-
-	getDetailVMInfoByGpuDevicesParams := clientgpu.NewGetDetailVMInfoByGpuDevicesParams()
-	getDetailVMInfoByGpuDevicesParams.RequestBody = &models.GetGpuDevicesRequestBody{
-		Where: &models.GpuDeviceWhereInput{},
+		getDetailVMInfoByGpuDevicesParams.RequestBody.Where.AvailableVgpusNumGt = TowerInt32(0)
 	}
 
 	getDetailVMInfoByGpuDevicesResp, err := svr.Session.GpuDevice.GetDetailVMInfoByGpuDevices(getDetailVMInfoByGpuDevicesParams)