File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -44,9 +44,7 @@ def _is_nvfatbin_available():
4444 return False
4545
4646
47- nvfatbin_available = pytest .mark .skipif (
48- not _is_nvfatbin_available (), reason = "nvfatbin bindings not available"
49- )
47+ nvfatbin_available = pytest .mark .skipif (not _is_nvfatbin_available (), reason = "nvfatbin bindings not available" )
5048
5149
5250@pytest .fixture (scope = "module" )
@@ -126,9 +124,7 @@ def get_saxpy_fatbin(init_cuda):
126124 sym_map = mod .symbol_mapping
127125
128126 # Compile to PTX targeting the second arch
129- ptx_mod = Program (
130- SAXPY_KERNEL , code_type = "c++" , options = ProgramOptions (arch = f"sm_{ second_arch } " )
131- ).compile (
127+ ptx_mod = Program (SAXPY_KERNEL , code_type = "c++" , options = ProgramOptions (arch = f"sm_{ second_arch } " )).compile (
132128 "ptx" ,
133129 name_expressions = ("saxpy<float>" , "saxpy<double>" ),
134130 )
You can’t perform that action at this time.
0 commit comments