Specialize blocked root boundary mode at compile time

2026-10-21 18:35:27 +08:00
parent fce24a64e8
commit 3a374faace
13 changed files with 127 additions and 39 deletions
@@ -27,7 +27,7 @@ def state(ny, nx):

 def compile_module(cfg, context, source):
    cfg.validate()
-    build = compiler.generate_config(cfg)
+    build = compiler.generate_config(cfg, block_root_bc=2 if source == "blocks.cu" else 0)
    try:
        module = compiler.load_module(compiler.compile_kernel(context.sm_arch, source=source, build_dir=build))
    finally: