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:
30
30
31
31
.PHONY : test-python
32
32
test-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
35
35
$(PYTEST ) -s -n 8 python/test/unit/language/test_subprocess.py
36
36
$(PYTEST ) -s -n 8 python/test/unit/test_debug.py --forked
37
37
$(PYTEST ) -s -n 8 python/triton_kernels/tests/
38
38
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
41
41
TRITON_ALWAYS_COMPILE=1 TRITON_DISABLE_LINE_INFO=0 LLVM_PASS_PLUGIN_PATH=python/triton/instrumentation/libGPUInstrumentationTestLib.so \
42
42
$(PYTEST ) --capture=tee-sys -rfs -vvv python/test/unit/instrumentation/test_gpuhello.py
43
43
You can’t perform that action at this time.
0 commit comments