fix

intel · Feb 25, 2025 · 1b622ce · 1b622ce
1 parent fcc9fd1
commit 1b622ce
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/.github/actions/inductor-xpu-e2e-test/action.yml b/.github/actions/inductor-xpu-e2e-test/action.yml
@@ -50,7 +50,11 @@ runs:
       if: ${{ inputs.env_prepare }}
       shell: bash
       run: |
-        source ${HOME}/mengfeil/myvenv-e2e/bin/activate
+        if [ "${{ inputs.driver }}" == "rolling" ];then
+          source activate e2e_ci
+        else
+          source ${HOME}/mengfeil/myvenv-e2e/bin/activate
+        fi
         source .github/scripts/env.sh ${{ inputs.pytorch }}
         if [[ ${{ inputs.suite }} == *"torchbench"* ]]; then
           if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then
@@ -94,7 +98,11 @@ runs:
         DisableScratchPages: ${{ inputs.driver == 'rolling' && '1' || '0' }}
       shell: bash
       run: |
-        source ${HOME}/mengfeil/myvenv-e2e/bin/activate
+        if [ "${{ inputs.driver }}" == "rolling" ];then
+          source activate e2e_ci
+        else
+          source ${HOME}/mengfeil/myvenv-e2e/bin/activate
+        fi
         source .github/scripts/env.sh ${{ inputs.pytorch }}
         cp .github/scripts/inductor_xpu_test.sh ../pytorch
         cd ../pytorch
@@ -162,7 +170,11 @@ runs:
           cat $var >> inductor_log/summary_accuracy.csv
         done
 
-        source ${HOME}/mengfeil/myvenv-e2e/bin/activate
+        if [ "${{ inputs.driver }}" == "rolling" ];then
+          source activate e2e_ci
+        else
+          source ${HOME}/mengfeil/myvenv-e2e/bin/activate
+        fi
         cd ${{ github.workspace }}
         cp .github/scripts/inductor_summary.py ../pytorch
         cd ../pytorch