From 3efecc28e6d9a627a991d69f4fb1f4d9ef1b0109 Mon Sep 17 00:00:00 2001 From: Olli Lupton Date: Tue, 26 Nov 2024 15:17:09 +0000 Subject: [PATCH] fix triton test execution --- .github/workflows/_ci.yaml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml index a8ec3d079..c0d6b89c7 100644 --- a/.github/workflows/_ci.yaml +++ b/.github/workflows/_ci.yaml @@ -458,8 +458,9 @@ jobs: docker run -i --shm-size=1g --gpus all --volume $PWD:/output \ ${{ needs.build-triton.outputs.DOCKER_TAG_FINAL }} \ bash <<"EOF" |& tee test-triton.log - # autotuner tests from jax-triton now hit a triton code path that uses utilities from pytorch... - pip install --no-deps torch --index-url https://download.pytorch.org/whl/cpu + # autotuner tests from jax-triton now hit a triton code path that uses utilities from pytorch; this relies on + # actually having a CUDA backend for pytoch + pip install --no-deps torch python /opt/jax-triton/tests/triton_call_test.py --xml_output_file /output/triton_test.xml EOF STATISTICS_SCRIPT: |