nits

Author: leofang

cuda_core/cuda/core/experimental/_program.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
@@ -403,6 +403,8 @@ def close(self):
     def __init__(self, code, code_type, options: ProgramOptions = None):
         self._mnff = Program._MembersNeededForFinalize(self, None)
 
+        self._options = options = check_or_create_options(ProgramOptions, options, "Program options")
+
         if code_type not in self._supported_code_type:
             raise NotImplementedError
 
@@ -416,8 +418,6 @@ def __init__(self, code, code_type, options: ProgramOptions = None):
         else:
             raise NotImplementedError
 
-        self._options = options = check_or_create_options(ProgramOptions, options, "Program options")
-
     def close(self):
         """Destroy this program."""
         self._mnff.close()

cuda_core/examples/vector_add.py

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

cuda_core/tests/test_linker.py

@@ -86,14 +86,10 @@ def test_linker_init(compile_ptx_functions, options):
 
 
 def test_linker_init_invalid_arch(compile_ptx_functions):
-    if culink_backend:
-        with pytest.raises(AttributeError):
-            options = LinkerOptions(arch="99", ptx=True)
-            Linker(*compile_ptx_functions, options=options)
-    else:
-        with pytest.raises(nvjitlink.nvJitLinkError):
-            options = LinkerOptions(arch="99", ptx=True)
-            Linker(*compile_ptx_functions, options=options)
+    err = AttributeError if culink_backend else nvjitlink.nvJitLinkError
+    with pytest.raises(err):
+        options = LinkerOptions(arch="99", ptx=True)
+        Linker(*compile_ptx_functions, options=options)
 
 
 @pytest.mark.skipif(culink_backend, reason="culink does not support ptx option")