Skip to content

Commit cde556f

Browse files
authored
Fix cuda device query (#7207)
1 parent 41edb3c commit cde556f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cli/src/pcluster/resources/imagebuilder/parallelcluster_test.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,7 @@ phases:
252252
echo "Correctly installed CUDA ${cuda_output}"
253253
254254
echo "Testing CUDA with deviceQuery..."
255-
if [ {{ test.OSArchitecture.outputs.stdout }} != 'arm64' ]; then
255+
if [ -f /usr/local/cuda-${cuda_ver}/extras/demo_suite/deviceQuery ]; then
256256
/usr/local/cuda-${cuda_ver}/extras/demo_suite/deviceQuery | grep -o "Result = PASS"
257257
[[ $? -ne 0 ]] && echo "CUDA deviceQuery test failed" && exit 1
258258
else

0 commit comments

Comments
 (0)