Update inductor_xpu_test.sh: don't add --freezing in reference mode; add new mode: inference-with-freezing

.github/workflows/e2e-accuracy.yml

@@ -18,12 +18,12 @@ on:
           - torchbench
         default: all
       mode:
-        description: Inference, inference-no-freezing, or training
+        description: Inference, inference-with-freezing, or training
         type: choice
         options:
           - all
           - inference
-          - inference-no-freezing
+          - inference-with-freezing
           - training
         default: all
       dtype:
@@ -86,7 +86,7 @@ jobs:
             suite='["${{ inputs.suite }}"]'
           fi
           if [[ -z "${{ inputs.mode }}" || "${{ inputs.mode }}" == "all" ]]; then
-            mode='["inference", "inference-no-freezing", "training"]'
+            mode='["inference", "inference-with-freezing", "training"]'
           else
             mode='["${{ inputs.mode }}"]'
           fi

.github/workflows/e2e-performance.yml

@@ -18,12 +18,12 @@ on:
           - torchbench
         default: all
       mode:
-        description: Inference, inference-no-freezing, or training
+        description: Inference, inference-with-freezing, or training
         type: choice
         options:
           - all
           - inference
-          - inference-no-freezing
+          - inference-with-freezing
           - training
         default: all
       dtype:
@@ -87,7 +87,7 @@ jobs:
             suite='["${{ inputs.suite }}"]'
           fi
           if [[ -z "${{ inputs.mode }}" || "${{ inputs.mode }}" == "all" ]]; then
-            mode='["inference", "inference-no-freezing", "training"]'
+            mode='["inference", "inference-with-freezing", "training"]'
           else
             mode='["${{ inputs.mode }}"]'
           fi

.github/workflows/e2e-reusable.yml

@@ -12,7 +12,7 @@ on:
         type: string
         default: all
       mode:
-        description: Inference, inference-no-freezing, or training
+        description: Inference, inference-with-freezing, or training
         type: string
         default: all
       test_mode:

.github/workflows/e2e-windows.yml

@@ -14,12 +14,12 @@ on:
           - torchbench
         default: all
       mode:
-        description: Inference, inference-no-freezing, or training
+        description: Inference, inference-with-freezing, or training
         type: choice
         options:
           - all
           - inference
-          - inference-no-freezing
+          - inference-with-freezing
           - training
         default: all
       dtype:
@@ -90,7 +90,7 @@ jobs:
             suite='["${{ inputs.suite }}"]'
           fi
           if [[ -z "${{ inputs.mode }}" || "${{ inputs.mode }}" == "all" ]]; then
-            mode='["inference", "inference-no-freezing", "training"]'
+            mode='["inference", "inference-with-freezing", "training"]'
           else
             mode='["${{ inputs.mode }}"]'
           fi

scripts/check_inductor_report.py

@@ -70,7 +70,7 @@ def main():
     argparser = argparse.ArgumentParser()
     argparser.add_argument("--suite", required=True)
     argparser.add_argument("--dtype", required=True)
-    argparser.add_argument("--mode", required=True, choices=("inference", "training", "inference-no-freezing"))
+    argparser.add_argument("--mode", required=True, choices=("inference", "training", "inference-with-freezing"))
     argparser.add_argument("--test_mode", required=True, choices=("performance", "accuracy"))
     argparser.add_argument("--device", help="i.e. xpu", required=True)
     argparser.add_argument("--models-file", help="Subset of models list", required=True)

scripts/inductor_xpu_test.sh

@@ -24,11 +24,13 @@ if [[ -n "$MODEL_ONLY" ]]; then
 fi
 
 if [[ $MODE == "inference" ]]; then
-    Mode_extra="--inference --freezing "
+    # For PT >= 2.1
+    # Remove --freezing cause feature not ready
+    Mode_extra="--inference"
 fi
 
-if [[ $MODE == "inference-no-freezing" ]]; then
-    Mode_extra="--inference "
+if [[ $MODE == "inference-with-freezing" ]]; then
+    Mode_extra="--inference --freezing "
 fi
 
 if [[ $MODE == "training" ]]; then