Skip to content

Commit 3c8f899

Browse files
committed
add NDEBUG option to CUTLASS compile to speed up attention kernel
1 parent e08caef commit 3c8f899

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

python/tvm/contrib/cutlass/build.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ def _get_cutlass_compile_options(sm, threads, use_fast_math=False):
5959
"-c",
6060
"-DCUTLASS_ENABLE_TENSOR_CORE_MMA=1",
6161
"-gencode=arch=compute_%d,code=[sm_%d,compute_%d]" % (sm, sm, sm),
62+
"-DNDEBUG",
6263
"-Xcompiler=-fPIC",
6364
"-Xcompiler=-Wconversion",
6465
"-Xcompiler=-fno-strict-aliasing",

0 commit comments

Comments
 (0)