Enable DRA Partitionable Devices support in kueue-operator CI

ci-operator/config/openshift/kueue-operator/openshift-kueue-operator-main.yaml

@@ -145,6 +145,8 @@ tests:
 - as: test-e2e-ci-build-upstream-4-21
   steps:
     cluster_profile: openshift-org-aws
+    env:
+      DRA_GPU_PARTITIONS: "4"
     post:
     - chain: kueue-operator-post
     test:
@@ -170,7 +172,7 @@ tests:
       cli: latest
       commands: make e2e-upstream-test
       env:
-      - default: singlecluster certmanager dra
+      - default: singlecluster certmanager sequential dra/whole-device dra/counter
         name: E2E_TARGET_FOLDERS
       from: kueue-operator-src
       resources:

ci-operator/config/openshift/kueue-operator/openshift-kueue-operator-release-1.4.yaml

@@ -145,6 +145,8 @@ tests:
 - as: test-e2e-ci-build-upstream-4-21
   steps:
     cluster_profile: openshift-org-aws
+    env:
+      DRA_GPU_PARTITIONS: "4"
     post:
     - chain: kueue-operator-post
     test:
@@ -170,7 +172,7 @@ tests:
       cli: latest
       commands: make e2e-upstream-test
       env:
-      - default: singlecluster certmanager dra
+      - default: singlecluster certmanager sequential dra/whole-device dra/counter
         name: E2E_TARGET_FOLDERS
       from: kueue-operator-src
       resources:

ci-operator/step-registry/dra-example-driver/install/dra-example-driver-install-commands.sh

@@ -70,11 +70,18 @@ echo ""
 
 # Install dra-example-driver via Helm from local chart
 echo "Installing dra-example-driver ${DRA_EXAMPLE_DRIVER_VERSION}..."
+GPU_PARTITION_ARGS=""
+if [ -n "${DRA_GPU_PARTITIONS:-}" ]; then
+  echo "GPU partitions: ${DRA_GPU_PARTITIONS}"
+  GPU_PARTITION_ARGS="--set gpu.partitions=${DRA_GPU_PARTITIONS}"
+fi
+
 helm upgrade --install \
   --namespace "${DRA_EXAMPLE_DRIVER_NAMESPACE}" \
   dra-example-driver \
   "${DRA_CHART_DIR}" \
   --set kubeletPlugin.containers.plugin.securityContext.privileged=true \
+  ${GPU_PARTITION_ARGS} \
   --wait \
   --timeout 10m
 

ci-operator/step-registry/dra-example-driver/install/dra-example-driver-install-ref.yaml

@@ -9,10 +9,16 @@ ref:
       memory: 200Mi
   env:
   - name: DRA_EXAMPLE_DRIVER_VERSION
-    default: "v0.2.1"
+    default: "v0.3.0"
     documentation: |-
       Version of dra-example-driver to install via Helm.
       Check https://github.com/kubernetes-sigs/dra-example-driver/releases for available versions.
+  - name: DRA_GPU_PARTITIONS
+    default: ""
+    documentation: |-
+      Number of GPU partitions for dra-example-driver.
+      When set, passes --set gpu.partitions=<value> to the Helm install.
+      Required for Partitionable Devices (counter-based) DRA tests.
   documentation: |-
     Installs the dra-example-driver from kubernetes-sigs/dra-example-driver
     via Helm from a local chart extracted from the source tarball.

ci-operator/step-registry/kueue-operator/test/e2e/dra/enable-feature-gate/kueue-operator-test-e2e-dra-enable-feature-gate-commands.sh

@@ -4,14 +4,14 @@ set -o errexit
 set -o pipefail
 
 echo "========================================="
-echo "Enabling DRAExtendedResource feature gate"
+echo "Enabling DRA feature gates"
 echo "========================================="
 echo ""
 
 export KUBECONFIG="${SHARED_DIR}/kubeconfig"
 
 echo "Patching featuregates..."
-oc patch featuregates cluster --type='merge' -p '{"spec":{"featureSet":"CustomNoUpgrade","customNoUpgrade":{"enabled":["DRAExtendedResource"]}}}'
+oc patch featuregates cluster --type='merge' -p '{"spec":{"featureSet":"CustomNoUpgrade","customNoUpgrade":{"enabled":["DRAExtendedResource","DRAPartitionableDevices"]}}}'
 
 echo "Waiting for kube-apiserver to start rolling out..."
 oc wait co kube-apiserver --for='condition=Progressing=True' --timeout=5m || true
@@ -20,5 +20,5 @@ echo "Waiting for kube-apiserver rollout to complete..."
 oc wait co kube-apiserver --for='condition=Progressing=False' --timeout=30m
 
 echo ""
-echo "DRAExtendedResource feature gate enabled"
+echo "DRA feature gates enabled (DRAExtendedResource, DRAPartitionableDevices)"
 echo ""