From 5b897f3570da4913d332acf57c648f6cea9beaa4 Mon Sep 17 00:00:00 2001 From: Karthik Vetrivel Date: Fri, 4 Sep 2026 13:53:51 -0700 Subject: [PATCH] Set extendedResourceName on gpu.nvidia.com DeviceClass for resource.k8s.io/v1 Signed-off-by: Karthik Vetrivel --- .../state/testdata/golden/gpucluster-dra-driver-full-spec.yaml | 1 + .../state/testdata/golden/gpucluster-dra-driver-gpus-only.yaml | 1 + .../golden/gpucluster-dra-driver-passthrough-support.yaml | 1 + manifests/state-dra-driver/0400_deviceclass-gpu.yaml | 3 +++ 4 files changed, 6 insertions(+) diff --git a/internal/state/testdata/golden/gpucluster-dra-driver-full-spec.yaml b/internal/state/testdata/golden/gpucluster-dra-driver-full-spec.yaml index f3552d5d2a..bdc760981c 100644 --- a/internal/state/testdata/golden/gpucluster-dra-driver-full-spec.yaml +++ b/internal/state/testdata/golden/gpucluster-dra-driver-full-spec.yaml @@ -343,6 +343,7 @@ kind: DeviceClass metadata: name: gpu.nvidia.com spec: + extendedResourceName: nvidia.com/gpu selectors: - cel: expression: device.driver == 'gpu.nvidia.com' && device.attributes['gpu.nvidia.com'].type diff --git a/internal/state/testdata/golden/gpucluster-dra-driver-gpus-only.yaml b/internal/state/testdata/golden/gpucluster-dra-driver-gpus-only.yaml index 5f371e7f74..59e436dd6b 100644 --- a/internal/state/testdata/golden/gpucluster-dra-driver-gpus-only.yaml +++ b/internal/state/testdata/golden/gpucluster-dra-driver-gpus-only.yaml @@ -136,6 +136,7 @@ kind: DeviceClass metadata: name: gpu.nvidia.com spec: + extendedResourceName: nvidia.com/gpu selectors: - cel: expression: device.driver == 'gpu.nvidia.com' && device.attributes['gpu.nvidia.com'].type diff --git a/internal/state/testdata/golden/gpucluster-dra-driver-passthrough-support.yaml b/internal/state/testdata/golden/gpucluster-dra-driver-passthrough-support.yaml index 99df54e1b0..9ef6f27467 100644 --- a/internal/state/testdata/golden/gpucluster-dra-driver-passthrough-support.yaml +++ b/internal/state/testdata/golden/gpucluster-dra-driver-passthrough-support.yaml @@ -136,6 +136,7 @@ kind: DeviceClass metadata: name: gpu.nvidia.com spec: + extendedResourceName: nvidia.com/gpu selectors: - cel: expression: device.driver == 'gpu.nvidia.com' && device.attributes['gpu.nvidia.com'].type diff --git a/manifests/state-dra-driver/0400_deviceclass-gpu.yaml b/manifests/state-dra-driver/0400_deviceclass-gpu.yaml index 3b11494a35..a248c8d42d 100644 --- a/manifests/state-dra-driver/0400_deviceclass-gpu.yaml +++ b/manifests/state-dra-driver/0400_deviceclass-gpu.yaml @@ -6,3 +6,6 @@ spec: selectors: - cel: expression: "device.driver == 'gpu.nvidia.com' && device.attributes['gpu.nvidia.com'].type == 'gpu'" +{{- if eq .DeviceClassAPIVersion "resource.k8s.io/v1" }} + extendedResourceName: nvidia.com/gpu +{{- end }}