diff --git a/.github/workflows/e2e-accuracy.yml b/.github/workflows/e2e-accuracy.yml index 335568f6bd..513fdb6b72 100644 --- a/.github/workflows/e2e-accuracy.yml +++ b/.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 diff --git a/.github/workflows/e2e-performance.yml b/.github/workflows/e2e-performance.yml index c8dc328b7f..6c074093c1 100644 --- a/.github/workflows/e2e-performance.yml +++ b/.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 diff --git a/.github/workflows/e2e-reusable.yml b/.github/workflows/e2e-reusable.yml index 6c56cdd0ce..bdee34ec40 100644 --- a/.github/workflows/e2e-reusable.yml +++ b/.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: diff --git a/.github/workflows/e2e-windows.yml b/.github/workflows/e2e-windows.yml index 9d06318662..ab9940d9cb 100644 --- a/.github/workflows/e2e-windows.yml +++ b/.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 diff --git a/scripts/check_inductor_report.py b/scripts/check_inductor_report.py index 9a14f840b6..47ea25e2b8 100755 --- a/scripts/check_inductor_report.py +++ b/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) diff --git a/scripts/inductor_xpu_test.sh b/scripts/inductor_xpu_test.sh index 7373f6506c..b039507d8d 100755 --- a/scripts/inductor_xpu_test.sh +++ b/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