Make bootstrap GPU helper container runtime aware (awslabs#1858)

mebays · Jul 26, 2024 · ed70934 · ed70934
1 parent 34f466e
commit ed70934
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/templates/al2/runtime/bootstrap.sh b/templates/al2/runtime/bootstrap.sh
@@ -639,7 +639,7 @@ fi
 BOOTSTRAP_GPU_HELPER=/etc/eks/bootstrap-gpu.sh
 if [ -x "${BOOTSTRAP_GPU_HELPER}" ]; then
   log "INFO: starting GPU bootstrap helper..."
-  "${BOOTSTRAP_GPU_HELPER}"
+  $BOOTSTRAP_GPU_HELPER -r "$CONTAINER_RUNTIME"
   log "INFO: completed GPU bootstrap helper!"
 fi