D1.4: Add PtxasOptions -j64 + OptLevel(0) for faster hd=512 compilation

This commit is contained in:
2026-05-24 15:36:35 +00:00
parent 331ddb29b7
commit 4f69dffc93

View File

@@ -45,7 +45,11 @@ def test():
import time
t0 = time.time()
compiled = cute.compile(kernel, mQ, mK, mV, mC, stream, mLSE)
from cutlass.base_dsl.compiler import CompileOptions, PtxasOptions, OptLevel
# PtxasOptions -j64: use 64 threads for ptxas register allocation (B200 has 256 cores)
# OptLevel(0): skip MLIR optimizations for faster compilation (first verify correctness, then optimize)
compile_opts = CompileOptions((PtxasOptions("-j64"), OptLevel(0)))
compiled = cute.compile(kernel, mQ, mK, mV, mC, stream, mLSE, config=compile_opts)
t1 = time.time()
print(f'Compilation took {t1-t0:.1f}s', flush=True)