update

beam-cloud · Jan 3, 2025 · ca454cc · ca454cc
1 parent ac65703
commit ca454cc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/worker/cr.go b/pkg/worker/cr.go
@@ -187,8 +187,10 @@ func (s *Worker) waitForRestoredContainer(ctx context.Context, containerId strin
 		return exitCode
 	}
 
-	go s.collectAndSendContainerMetrics(ctx, request, spec, pid) // Capture resource usage (cpu/mem/gpu)
-	go s.watchOOMEvents(ctx, containerId, outputLogger)          // Watch for OOM events
+	gpuDeviceIds, _ := s.containerCudaManager.GetContainerGPUDevices(containerId)
+
+	go s.collectAndSendContainerMetrics(ctx, request, spec, pid, gpuDeviceIds) // Capture resource usage (cpu/mem/gpu)
+	go s.watchOOMEvents(ctx, containerId, outputLogger)                        // Watch for OOM events
 
 	ticker := time.NewTicker(1 * time.Second)
 	defer ticker.Stop()