File tree Expand file tree Collapse file tree 1 file changed +4
-9
lines changed Expand file tree Collapse file tree 1 file changed +4
-9
lines changed Original file line number Diff line number Diff line change @@ -374,7 +374,7 @@ jobs:
374374 path : ${{ github.workspace }}/ut_log
375375
376376 distributed_ut_test :
377- runs-on : pytorch-06
377+ runs-on : PVC-7358
378378 if : ${{ contains(inputs.ut, 'xpu_distributed') && !contains(inputs.disabled_tests, 'disable_distribute') }}
379379 timeout-minutes : 60
380380 env :
@@ -447,13 +447,9 @@ jobs:
447447 fi
448448 if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then
449449 pip install cmake ninja pybind11
450- rm -rf pytorch_triton_xpu-*.whl
451- TRITON_VERSION_NAME="$(
452- curl -sSL https://raw.githubusercontent.com/intel/intel-xpu-backend-for-triton/${TRITON_COMMIT_ID}/python/triton/__init__.py 2>&1 |\
453- grep '__version__' |head -n 1 |awk -F "'" '{print $2}'
454- )"
455- python .github/scripts/build_triton_wheel.py --device xpu --commit-hash ${TRITON_COMMIT_ID} --triton-version ${TRITON_VERSION_NAME}
456- pip install pytorch_triton_xpu-*.whl
450+ TRITON_REPO="https://github.com/intel/intel-xpu-backend-for-triton"
451+ TRITON_COMMIT_ID="bdd0656b"
452+ pip install --force-reinstall "git+${TRITON_REPO}@${TRITON_COMMIT_ID}#subdirectory=python"
457453 fi
458454 - name : Torch Config
459455 run : |
@@ -491,7 +487,6 @@ jobs:
491487 2>${{ github.workspace }}/ut_log/xpu_distributed/pytorch_distributed_test_error.log | \
492488 tee ${{ github.workspace }}/ut_log/xpu_distributed/pytorch_distributed_test.log
493489 cp *.xml ${{ github.workspace }}/ut_log
494- cp
495490 - name : Reset Ptrace_scope
496491 if : ${{ always() }}
497492 run : |
You can’t perform that action at this time.
0 commit comments