@@ -138,6 +138,12 @@ jobs:
138138 image_key : test_py312
139139 - python-version : " 3.13"
140140 image_key : test_py313
141+ container :
142+ image : ${{ needs.images.outputs[matrix.image_key] }}
143+ credentials :
144+ username : ${{ github.actor }}
145+ password : ${{ secrets.GITHUB_TOKEN }}
146+ options : --gpus all
141147 steps :
142148 - name : Checkout repository
143149 uses : actions/checkout@v6
@@ -148,27 +154,15 @@ jobs:
148154 name : wheel-py${{ matrix.python-version }}-linux-x86_64
149155 path : dist/
150156
151- - name : Log in to GHCR
152- uses : docker/login-action@v3
153- with :
154- registry : ghcr.io
155- username : ${{ github.actor }}
156- password : ${{ secrets.GITHUB_TOKEN }}
157-
158- - name : Pull test image
159- run : docker pull ${{ needs.images.outputs[matrix.image_key] }}
157+ - name : Install wheel
158+ run : pip install dist/*.whl
160159
161160 - name : Run tests
162161 run : |
163- docker run --rm --gpus all \
164- -v "${{ github.workspace }}":/workspace \
165- -w /workspace \
166- ${{ needs.images.outputs[matrix.image_key] }} \
167- bash -c "pip install dist/*.whl && \
168- pytest --ignore internal \
169- -m 'not benchmark and not use_mlir' \
170- --durations=10 \
171- --junitxml=/workspace/test-results.xml"
162+ pytest --ignore internal \
163+ -m 'not benchmark and not use_mlir' \
164+ --durations=10 \
165+ --junitxml=test-results.xml
172166
173167 - name : Upload test results
174168 if : always()
@@ -185,7 +179,3 @@ jobs:
185179 name : Test Results (Python ${{ matrix.python-version }})
186180 path : test-results.xml
187181 reporter : java-junit
188-
189- - name : Fix workspace permissions
190- if : always()
191- run : sudo chown -R $(id -u):$(id -g) "${{ github.workspace }}"
0 commit comments