Skip to content

Commit 773e22f

Browse files
[pre-commit.ci] auto code formatting
1 parent 544065b commit 773e22f

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

cuda_core/tests/test_module.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff 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
)

0 commit comments

Comments
 (0)