biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

D1.4: Use options string for compile flags (--ptxas-options -j64 --opt-level 0)

Browse Source
This commit is contained in:
biondizzle
2026-05-24 15:40:39 +00:00
parent 8771f11fa0
commit 4bb34ea3b3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unit/test_d1_hd512_only.py
View File

@@ -47,9 +47,9 @@ def test():
t0 = time.time()
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)
# OptLevel(0): skip MLIR optimizations for faster compilation (verify correctness first, then optimize)
compiled = cute.compile(kernel, mQ, mK, mV, mC, stream, mLSE,
options="--ptxas-options '-j64' --opt-level 0")
t1 = time.time()
print(f'Compilation took {t1-t0:.1f}s', flush=True)