Skip to content

Commit b7fe821

Browse files
committed
Add device plugin options to registration method
Signed-off-by: ghokun <[email protected]>
1 parent 97bcad9 commit b7fe821

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

manifests/device-plugin.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ spec:
4545
capabilities:
4646
add: ["SYS_ADMIN"]
4747
containers:
48-
- image: ghcr.io/kuartis/kuartis-virtual-gpu-device-plugin:0.6.1
48+
- image: ghcr.io/kuartis/kuartis-virtual-gpu-device-plugin:0.6.2
4949
name: kuartis-virtual-gpu-device-plugin-ctr
5050
command:
5151
- /usr/bin/virtual-gpu-device-plugin

pkg/gpu/nvidia/server.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,9 @@ func (m *NvidiaDevicePlugin) Register(kubeletEndpoint, resourceName string) erro
163163
Version: pluginapi.Version,
164164
Endpoint: path.Base(m.socket),
165165
ResourceName: resourceName,
166+
Options: &pluginapi.DevicePluginOptions{
167+
GetPreferredAllocationAvailable: !m.allowMultiGpu,
168+
},
166169
}
167170

168171
_, err = client.Register(context.Background(), reqt)

0 commit comments

Comments
 (0)