fixes

Author: metascroy

.ci/scripts/validate.sh

@@ -322,16 +322,16 @@ CHECKPOINT_PATH="$1"
 TARGET_DEVICE="${2:-cpu}"
 PROMPT="Hello, my name is"
 
+DTYPES="${3:-default}"
 
-if [ "$#" -gt 2 ]; then
+if [ "$#" -gt 3 ]; then
     # Additional arguments provided
     for arg in "${@:3}"; do
         case "$arg" in
             "compile")
                 run_compile || exit 1
                 ;;
             "aoti")
-                DTYPES="${4:-default}"
                 run_aoti || exit 1
                 ;;
             "executorch")

.github/workflows/pull.yml

@@ -185,7 +185,7 @@ jobs:
 
   test-gpu-aoti-bfloat16:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
-    name: test-gpu-aoti (${{ matrix.platform }}, ${{ matrix.model_name }})
+    name: test-gpu-aoti-bfloat16 (${{ matrix.platform }}, ${{ matrix.model_name }})
     needs: gather-models-gpu
     timeout: 60
     strategy:
@@ -222,7 +222,7 @@ jobs:
 
   test-gpu-aoti-float32:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
-    name: test-gpu-aoti (${{ matrix.platform }}, ${{ matrix.model_name }})
+    name: test-gpu-aoti-float32 (${{ matrix.platform }}, ${{ matrix.model_name }})
     needs: gather-models-gpu
     strategy:
       matrix: ${{ fromJSON(needs.gather-models-gpu.outputs.models) }}
@@ -258,7 +258,7 @@ jobs:
 
   test-gpu-aoti-float16:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
-    name: test-gpu-aoti (${{ matrix.platform }}, ${{ matrix.model_name }})
+    name: test-gpu-aoti-float16 (${{ matrix.platform }}, ${{ matrix.model_name }})
     needs: gather-models-gpu
     strategy:
       matrix: ${{ fromJSON(needs.gather-models-gpu.outputs.models) }}