fix: update node resources according to gpu list (#68)

Author: 0x5457

cmd/nodediscovery/main.go

@@ -102,10 +102,10 @@ func main() {
 		os.Exit(1)
 	}
 
-	totalTFlops := resource.MustParse("0")
-	totalVRAM := resource.MustParse("0Ki")
-	availableTFlops := resource.MustParse("0")
-	availableVRAM := resource.MustParse("0Ki")
+	totalTFlops := resource.Quantity{}
+	totalVRAM := resource.Quantity{}
+	availableTFlops := resource.Quantity{}
+	availableVRAM := resource.Quantity{}
 
 	allDeviceIDs := make([]string, 0)
 
@@ -140,7 +140,7 @@ func main() {
 		})
 		tflops := info.Fp16TFlops
 		if !ok {
-			tflops = resource.MustParse("0")
+			tflops = resource.Quantity{}
 		}
 		gpu := &tfv1.GPU{
 			ObjectMeta: metav1.ObjectMeta{

internal/controller/gpunode_controller.go

@@ -208,6 +208,18 @@ func (r *GPUNodeReconciler) checkStatusAndUpdateVirtualCapacity(ctx context.Cont
 			return true, nil
 		}
 
+		node.Status.AvailableVRAM = resource.Quantity{}
+		node.Status.AvailableTFlops = resource.Quantity{}
+		node.Status.TotalTFlops = resource.Quantity{}
+		node.Status.TotalVRAM = resource.Quantity{}
+
+		for _, gpu := range gpuList {
+			node.Status.AvailableVRAM.Add(gpu.Status.Available.Vram)
+			node.Status.AvailableTFlops.Add(gpu.Status.Available.Tflops)
+			node.Status.TotalVRAM.Add(gpu.Status.Capacity.Vram)
+			node.Status.TotalTFlops.Add(gpu.Status.Capacity.Tflops)
+		}
+
 		virtualVRAM, virtualTFlops := r.CalculateVirtualCapacity(node, poolObj)
 		node.Status.VirtualTFlops = virtualTFlops
 		node.Status.VirtualVRAM = virtualVRAM