Skip to content

Commit c8a8490

Browse files
committed
update
1 parent 5ac5b25 commit c8a8490

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

.github/workflows/_linux_ut.yml

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff 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: |

0 commit comments

Comments
 (0)