Bug fix - handle fractions in status.queueDetails correctly(#61)

(r *ResourceRequirements) ToResourceList() will now handle fractional gpus correctly + unitest

Author: davidLif

pkg/scheduler/api/resource_info/resource_requirment.go

@@ -71,7 +71,12 @@ func RequirementsFromResourceList(rl v1.ResourceList) *ResourceRequirements {
 func (r *ResourceRequirements) ToResourceList() v1.ResourceList {
 	rl := r.BaseResource.ToResourceList()
 
-	rl[GPUResourceName] = *resource.NewQuantity(int64(r.GPUs()), resource.DecimalSI)
+	if r.GPUs() > 0 && r.GPUs() < wholeGpuPortion { // fractional GPUs
+		rl[GPUResourceName] = *resource.NewMilliQuantity(int64(r.GPUs()*1000), resource.DecimalSI)
+	} else { // whole GPUs
+		rl[GPUResourceName] = *resource.NewQuantity(r.GetNumOfGpuDevices(), resource.DecimalSI)
+	}
+
 	for rName, rQuant := range r.MigResources() {
 		rl[rName] = *resource.NewQuantity(rQuant, resource.DecimalSI)
 	}

pkg/scheduler/api/resource_info/resource_requirment_test.go

@@ -44,6 +44,14 @@ var _ = Describe("ResourceRequirements Info internal logic", func() {
 			newResourceList := resourceInfo.ToResourceList()
 			compareResourceLists(resourceList, newResourceList)
 		})
+		It("GPU Fractional Resources", func() {
+			resourceInfo := NewResourceRequirementsWithGpus(0.5)
+			Expect(resourceInfo.GPUs()).To(Equal(0.5))
+
+			newResourceList := resourceInfo.ToResourceList()
+			gpuResource := newResourceList[GPUResourceName]
+			Expect(gpuResource.MilliValue()).To(Equal(int64(500)))
+		})
 		It("Other Resources", func() {
 			resourceList := v1.ResourceList{
 				v1.ResourceName("run.ai/test-resource"): resource.MustParse("1"),