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:
374
374
path : ${{ github.workspace }}/ut_log
375
375
376
376
distributed_ut_test :
377
- runs-on : pytorch-06
377
+ runs-on : PVC-7358
378
378
if : ${{ contains(inputs.ut, 'xpu_distributed') && !contains(inputs.disabled_tests, 'disable_distribute') }}
379
379
timeout-minutes : 60
380
380
env :
@@ -447,13 +447,9 @@ jobs:
447
447
fi
448
448
if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then
449
449
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"
457
453
fi
458
454
- name : Torch Config
459
455
run : |
@@ -491,7 +487,6 @@ jobs:
491
487
2>${{ github.workspace }}/ut_log/xpu_distributed/pytorch_distributed_test_error.log | \
492
488
tee ${{ github.workspace }}/ut_log/xpu_distributed/pytorch_distributed_test.log
493
489
cp *.xml ${{ github.workspace }}/ut_log
494
- cp
495
490
- name : Reset Ptrace_scope
496
491
if : ${{ always() }}
497
492
run : |
You can’t perform that action at this time.
0 commit comments