File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -65,6 +65,8 @@ sed -i -e 's|-Werror||g' CMakeLists.txt
6565sed -i -e 's|\(LLVMAMDGPU.*\)|# \1|g' CMakeLists.txt
6666# Do not build tests
6767sed -i -e 's|^\s*add_subdirectory(unittest)|# unit tests disabled|' CMakeLists.txt
68+ # Avoid building GPUHello.cpp, which does not compile with cl693214397 of OpenXLA Triton
69+ sed -i -e 's|^add_subdirectory(Instrumentation)|# Instrumentation test disabled|' test/lib/CMakeLists.txt
6870# Avoid error due to forward declaration of Module
6971sed -i -e '/#include "llvm\/IR\/IRBuilder.h"/a #include "llvm/IR/Module.h"' test/lib/Instrumentation/GPUHello.cpp
7072# Google has patches that mess with include paths in source files
Original file line number Diff line number Diff line change @@ -458,8 +458,9 @@ jobs:
458458 docker run -i --shm-size=1g --gpus all --volume $PWD:/output \
459459 ${{ needs.build-triton.outputs.DOCKER_TAG_FINAL }} \
460460 bash <<"EOF" |& tee test-triton.log
461- # autotuner tests from jax-triton now hit a triton code path that uses utilities from pytorch...
462- pip install --no-deps torch --index-url https://download.pytorch.org/whl/cpu
461+ # autotuner tests from jax-triton now hit a triton code path that uses utilities from pytorch; this relies on
462+ # actually having a CUDA backend for pytoch
463+ pip install --no-deps torch
463464 python /opt/jax-triton/tests/triton_call_test.py --xml_output_file /output/triton_test.xml
464465 EOF
465466 STATISTICS_SCRIPT : |
You can’t perform that action at this time.
0 commit comments