Skip to content

Commit cb336b1

Browse files
committed
tests/ai-conformance: add toleration for nvidia.com/gpu taint
1 parent 2641897 commit cb336b1

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/e2e/scenarios/ai-conformance/run-test.sh

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,15 @@ helm upgrade -i nvidia-gpu-operator --wait \
135135
# Uses the driver installed by GPU Operator at /run/nvidia/driver (the default).
136136
echo "Installing NVIDIA DRA Driver..."
137137

138+
cat > values.yaml <<EOF
139+
# The driver daemonset needs a toleration for the nvidia.com/gpu taint
140+
kubeletPlugin:
141+
tolerations:
142+
- key: nvidia.com/gpu
143+
operator: Exists
144+
effect: NoSchedule
145+
EOF
146+
138147
helm upgrade -i nvidia-dra-driver-gpu nvidia/nvidia-dra-driver-gpu \
139148
--version="25.12.0" \
140149
--create-namespace \

0 commit comments

Comments
 (0)