diff --git a/.github/actions/inductor-xpu-e2e-test/action.yml b/.github/actions/inductor-xpu-e2e-test/action.yml index 07a8136bc..cc9b32207 100644 --- 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