Update cuda_core/examples/tma_replace_address.py

rparolin · leofang · cpcloud · commit ae86192e0523 · 2026-03-14T10:28:50.000-04:00
Co-authored-by: Leo Fang &lt;leo80042@gmail.com&gt;
diff --git a/cuda_core/examples/tma_replace_address.py b/cuda_core/examples/tma_replace_address.py
@@ -146,7 +146,7 @@
 prog = Program(
     code,
     code_type="c++",
-    options=ProgramOptions(std="c++17", arch=f"sm_{arch_str}"),
+    options=ProgramOptions(std="c++17", arch=f"sm_{dev.arch}"),
 )
 mod = prog.compile("cubin")
 ker = mod.get_kernel("tma_copy")

Original file line number	Diff line number	Diff line change
`@@ -146,7 +146,7 @@`
`146`	`146`	`prog = Program(`
`147`	`147`	`code,`
`148`	`148`	`code_type="c++",`
`149`		`- options=ProgramOptions(std="c++17", arch=f"sm_{arch_str}"),`
	`149`	`+ options=ProgramOptions(std="c++17", arch=f"sm_{dev.arch}"),`
`150`	`150`	`)`
`151`	`151`	`mod = prog.compile("cubin")`
`152`	`152`	`ker = mod.get_kernel("tma_copy")`