Skip to content

Commit 9519904

Browse files
committed
also turn on parallel cythonization
1 parent 8cbf40c commit 9519904

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

cuda_core/build_hooks.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,12 @@ def get_cuda_paths():
9191
for mod in module_names
9292
)
9393

94+
nthreads = int(os.environ.get("CUDA_PYTHON_PARALLEL_LEVEL", os.cpu_count() // 2))
95+
9496
global _extensions
95-
_extensions = cythonize(ext_modules, verbose=True, language_level=3, compiler_directives={"embedsignature": True})
97+
_extensions = cythonize(
98+
ext_modules, verbose=True, language_level=3, nthreads=nthreads, compiler_directives={"embedsignature": True}
99+
)
96100

97101
return _build_meta.build_wheel(wheel_directory, config_settings, metadata_directory)
98102

0 commit comments

Comments
 (0)