File tree Expand file tree Collapse file tree 2 files changed +8
-14
lines changed
Expand file tree Collapse file tree 2 files changed +8
-14
lines changed Original file line number Diff line number Diff 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 : |
Original file line number Diff line number Diff line change @@ -315,15 +315,9 @@ run_instrumentation_tests() {
315315run_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
You can’t perform that action at this time.
0 commit comments