Merge pull request #73 from oracle-quickstart/cloud-init-update

Update cloud init scripts

Author: arnaudfroidmont

files/oke-nvme-raid.sh

@@ -43,6 +43,12 @@ fi
 # Determine config for detected device count and RAID level
 count=${#devices[@]}; bs=4; chunk=256
 stride=$((chunk/bs)) # chunk size / block size
+
+# If only 1 device, force RAID level 0
+if [[ $count -eq 1 ]]; then
+  level=0
+fi
+
 eff_count=$count # $level == 0
 if [[ $level == 10 ]]; then eff_count=$((count/2)); fi
 if [[ $level == 5 ]]; then eff_count=$((count-1)); fi

files/oke-ubuntu-cloud-init.sh

@@ -8,6 +8,17 @@ else
     exit 1
 fi
 
+# Disable nvidia-imex.service for GB200 and GB300 shapes for Dynamic Resource Allocation (DRA) compatibility
+SHAPE=$(curl -H "Authorization: Bearer Oracle" -L http://169.254.169.254/opc/v2/instance/shape)
+if [[ "$SHAPE" == BM.GPU.GB200* ]] || [[ "$SHAPE" == BM.GPU.GB300* ]]; then
+    echo "Disabling nvidia-imex.service for shape: $SHAPE"
+    if systemctl list-unit-files | grep -q nvidia-imex.service; then
+        systemctl disable --now nvidia-imex.service && systemctl mask nvidia-imex.service
+    else
+        echo "nvidia-imex.service not found, skipping"
+    fi
+fi
+
 case "$ID" in
     ubuntu)
         echo "Detected Ubuntu"