Skip to content

Commit 227d9c1

Browse files
committed
make linter happy
1 parent 3ff5e94 commit 227d9c1

2 files changed

Lines changed: 8 additions & 4 deletions

File tree

cuda_core/cuda/core/experimental/_device.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1026,7 +1026,7 @@ def _get_current_context(self, check_consistency=False) -> driver.CUcontext:
10261026
err, ctx = driver.cuCtxGetCurrent()
10271027

10281028
# TODO: We want to just call this:
1029-
#_check_driver_error(err)
1029+
# _check_driver_error(err)
10301030
# but even the simplest success check causes 50-100 ns. Wait until we cythonize this file...
10311031
if ctx is None:
10321032
_check_driver_error(err)

cuda_core/setup.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,21 +9,25 @@
99
from setuptools import Extension, setup
1010
from setuptools.command.build_ext import build_ext as _build_ext
1111

12-
1312
# It seems setuptools' wildcard support has problems for namespace packages,
1413
# so we explicitly spell out all Extension instances.
1514
root_module = "cuda.core.experimental"
1615
root_path = f"{os.path.sep}".join(root_module.split(".")) + os.path.sep
1716
ext_files = glob.glob(f"{root_path}/**/*.pyx", recursive=True)
17+
18+
1819
def strip_prefix_suffix(filename):
19-
return filename[len(root_path):-4]
20+
return filename[len(root_path) : -4]
21+
22+
2023
module_names = (strip_prefix_suffix(f) for f in ext_files)
2124
ext_modules = tuple(
2225
Extension(
2326
f"cuda.core.experimental.{mod.replace(os.path.sep, '.')}",
2427
sources=[f"cuda/core/experimental/{mod}.pyx"],
2528
language="c++",
26-
) for mod in module_names
29+
)
30+
for mod in module_names
2731
)
2832

2933

0 commit comments

Comments
 (0)