Add logs for preferred allocation

Signed-off-by: ghokun <gokhun@gmail.com>
kuartis · Mar 24, 2022 · 97bcad9 · 97bcad9
1 parent a3cdfcb
commit 97bcad9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/manifests/device-plugin.yml b/manifests/device-plugin.yml
@@ -45,7 +45,7 @@ spec:
             capabilities:
               add: ["SYS_ADMIN"]
       containers:
-        - image: ghcr.io/kuartis/kuartis-virtual-gpu-device-plugin:0.6.0
+        - image: ghcr.io/kuartis/kuartis-virtual-gpu-device-plugin:0.6.1
           name: kuartis-virtual-gpu-device-plugin-ctr
           command:
             - /usr/bin/virtual-gpu-device-plugin

diff --git a/pkg/gpu/nvidia/server.go b/pkg/gpu/nvidia/server.go
@@ -201,6 +201,7 @@ func (m *NvidiaDevicePlugin) GetDevicePluginOptions(context.Context, *pluginapi.
 
 // GetPreferredAllocation returns the preferred allocation from the set of devices specified in the request
 func (m *NvidiaDevicePlugin) GetPreferredAllocation(ctx context.Context, r *pluginapi.PreferredAllocationRequest) (*pluginapi.PreferredAllocationResponse, error) {
+	log.Printf("GetPreferredAllocation request: %v", r)
 	response := &pluginapi.PreferredAllocationResponse{}
 	for _, req := range r.ContainerRequests {
 		// available vGPUs per physical GPU
@@ -224,6 +225,7 @@ func (m *NvidiaDevicePlugin) GetPreferredAllocation(ctx context.Context, r *plug
 			DeviceIDs: findAvailableDevicesOnSamePhysicalGPU(availableGpuMap, mustIncludeGpuMap, int(req.AllocationSize)),
 		})
 	}
+	log.Printf("GetPreferredAllocation response: %v", response)
 	return response, nil
 }