File tree Expand file tree Collapse file tree 3 files changed +715
-104
lines changed Expand file tree Collapse file tree 3 files changed +715
-104
lines changed Original file line number Diff line number Diff line change @@ -30,14 +30,14 @@ test-cpp:
3030
3131.PHONY : test-python
3232test-unit : all
33- cd python/test/unit && $(PYTEST ) -s -n 8 --ignore=language/test_line_info .py \
34- --ignore=language/test_subprocess.py --ignore=test_debug.py
33+ cd python/test/unit && $(PYTEST ) -s -n 8 --ignore=cuda/test_flashattention .py \
34+ --ignore=language/test_line_info.py --ignore=language/ test_subprocess.py --ignore=test_debug.py
3535 $(PYTEST ) -s -n 8 python/test/unit/language/test_subprocess.py
3636 $(PYTEST ) -s -n 8 python/test/unit/test_debug.py --forked
3737 $(PYTEST ) -s -n 8 python/triton_kernels/tests/
3838 TRITON_DISABLE_LINE_INFO=0 $(PYTEST ) -s python/test/unit/language/test_line_info.py
39- # Run attention separately to avoid out of gpu memory
40- $(PYTEST ) -s python/tutorials/06-fused-attention .py
39+ # Run cuda/test_flashattention.py separately to avoid out of gpu memory
40+ $(PYTEST ) -s python/test/unit/cuda/test_flashattention .py
4141 TRITON_ALWAYS_COMPILE=1 TRITON_DISABLE_LINE_INFO=0 LLVM_PASS_PLUGIN_PATH=python/triton/instrumentation/libGPUInstrumentationTestLib.so \
4242 $(PYTEST ) --capture=tee-sys -rfs -vvv python/test/unit/instrumentation/test_gpuhello.py
4343
You can’t perform that action at this time.
0 commit comments