Skip to content

Commit 5ac822d

Browse files
committed
Add a standalone e2e test, r7
1 parent 59bdfe0 commit 5ac822d

File tree

2 files changed

+8
-14
lines changed

2 files changed

+8
-14
lines changed

.github/workflows/build-test-reusable.yml

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -134,13 +134,13 @@ jobs:
134134
with:
135135
build_llvm: ${{ inputs.build_llvm }}
136136

137-
- name: Setup Triton
137+
- name: Setup Triton and Transformers
138138
if: inputs.env_manager != 'base'
139-
run: |
140-
export DEBUG=1
141-
cd python
142-
pip install pybind11
143-
pip install --no-build-isolation -e '.[build,tests,tutorials]'
139+
uses: ./.github/actions/install-wheels
140+
with:
141+
gh_token: ${{ secrets.GITHUB_TOKEN }}
142+
install_cmd: conda run --no-capture-output -n triton pip install
143+
python_version: ${{ matrix.python }}
144144

145145
- name: Create test-triton command line
146146
run: |

scripts/test-triton.sh

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -315,15 +315,9 @@ run_instrumentation_tests() {
315315
run_inductor_tests() {
316316
test -d pytorch || {
317317
git clone https://github.com/pytorch/pytorch
318+
rev=$(cat .github/pins/pytorch-upstream.txt)
318319
cd pytorch
319-
320-
TRANSFORMERS_VERSION=$(cat .ci/docker/ci_commit_pins/huggingface.txt)
321-
git clone --recursive https://github.com/huggingface/transformers
322-
cd transformers
323-
git checkout $TRANSFORMERS_VERSION
324-
python setup.py bdist_wheel
325-
pip install dist/*.whl
326-
python -c "import transformers; print(transformers.__version__)"
320+
git checkout $rev
327321
}
328322

329323
pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec

0 commit comments

Comments
 (0)