Inline dev.arch calls in f-strings per PR feedback

Co-authored-by: leofang <[email protected]>

Author: Copilot

cuda_core/examples/cuda_graphs.py

@@ -53,8 +53,7 @@ def main():
     cp.cuda.ExternalStream(int(stream.handle)).use()
 
     # Compile the program
-    arch = dev.arch
-    program_options = ProgramOptions(std="c++17", arch=f"sm_{arch}")
+    program_options = ProgramOptions(std="c++17", arch=f"sm_{dev.arch}")
     prog = Program(code, code_type="c++", options=program_options)
     mod = prog.compile(
         "cubin", name_expressions=("vector_add<float>", "vector_multiply<float>", "vector_subtract<float>")

cuda_core/examples/memory_ops.py

@@ -54,8 +54,7 @@
 cp.cuda.ExternalStream(int(stream.handle)).use()
 
 # Compile kernel
-arch = dev.arch
-program_options = ProgramOptions(std="c++17", arch=f"sm_{arch}")
+program_options = ProgramOptions(std="c++17", arch=f"sm_{dev.arch}")
 prog = Program(code, code_type="c++", options=program_options)
 mod = prog.compile("cubin")
 kernel = mod.get_kernel("memory_ops")

cuda_core/examples/pytorch_example.py

@@ -51,8 +51,7 @@ def __cuda_stream__(self):
 s = dev.create_stream(PyTorchStreamWrapper(pt_stream))
 
 # prepare program
-arch = dev.arch
-program_options = ProgramOptions(std="c++11", arch=f"sm_{arch}")
+program_options = ProgramOptions(std="c++11", arch=f"sm_{dev.arch}")
 prog = Program(code, code_type="c++", options=program_options)
 mod = prog.compile(
     "cubin",

cuda_core/examples/saxpy.py

@@ -38,8 +38,7 @@
 s = dev.create_stream()
 
 # prepare program
-arch = dev.arch
-program_options = ProgramOptions(std="c++11", arch=f"sm_{arch}")
+program_options = ProgramOptions(std="c++11", arch=f"sm_{dev.arch}")
 prog = Program(code, code_type="c++", options=program_options)
 
 # Note the use of the `name_expressions` argument to specify the template

cuda_core/examples/simple_multi_gpu_example.py

@@ -40,8 +40,7 @@
     }
 }
 """
-arch0 = dev0.arch
-prog_add = Program(code_add, code_type="c++", options={"std": "c++17", "arch": f"sm_{arch0}"})
+prog_add = Program(code_add, code_type="c++", options={"std": "c++17", "arch": f"sm_{dev0.arch}"})
 mod_add = prog_add.compile("cubin")
 ker_add = mod_add.get_kernel("vector_add")
 
@@ -63,8 +62,7 @@
     }
 }
 """
-arch1 = dev1.arch
-prog_sub = Program(code_sub, code_type="c++", options={"std": "c++17", "arch": f"sm_{arch1}"})
+prog_sub = Program(code_sub, code_type="c++", options={"std": "c++17", "arch": f"sm_{dev1.arch}"})
 mod_sub = prog_sub.compile("cubin")
 ker_sub = mod_sub.get_kernel("vector_sub")
 

cuda_core/examples/strided_memory_view_gpu.py

@@ -103,8 +103,7 @@ def run():
     # To know the GPU's compute capability, we need to identify which GPU to use.
     dev = Device(0)
     dev.set_current()
-    arch = dev.arch
-    gpu_prog = Program(gpu_code, code_type="c++", options=ProgramOptions(arch=f"sm_{arch}", std="c++11"))
+    gpu_prog = Program(gpu_code, code_type="c++", options=ProgramOptions(arch=f"sm_{dev.arch}", std="c++11"))
     mod = gpu_prog.compile(target_type="cubin")
     gpu_ker = mod.get_kernel(func_name)
 

cuda_core/examples/vector_add.py

@@ -33,8 +33,7 @@
 s = dev.create_stream()
 
 # prepare program
-arch = dev.arch
-program_options = ProgramOptions(std="c++17", arch=f"sm_{arch}")
+program_options = ProgramOptions(std="c++17", arch=f"sm_{dev.arch}")
 prog = Program(code, code_type="c++", options=program_options)
 mod = prog.compile("cubin", name_expressions=("vector_add<float>",))