Skip to content

Commit ac735cf

Browse files
style
1 parent e7353bc commit ac735cf

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

numba_cuda/numba/cuda/cuda_paths.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def get_debian_pkg_libdevice():
262262

263263
def get_current_cuda_target_name():
264264
"""Determine conda's CTK target folder based on system and machine arch.
265-
265+
266266
CTK's conda package delivers headers based on its architecture type. For example,
267267
`x86_64` machine places header under `$CONDA_PREFIX/targets/x86_64-linux`, and
268268
`aarch64` places under `$CONDA_PREFIX/targets/sbsa-linux`. Read more about the
@@ -278,9 +278,10 @@ def get_current_cuda_target_name():
278278
'aarch64': 'sbsa-linux'
279279
}
280280
return arch_to_targets.get(machine)
281-
281+
282282
return None
283283

284+
284285
def get_conda_include_dir():
285286
"""
286287
Return the include directory in the current conda environment, if one
@@ -291,10 +292,13 @@ def get_conda_include_dir():
291292

292293
if conda_prefix:
293294
if target_name:
294-
include_dir = os.path.join(conda_prefix, f'targets/{target_name}/include')
295+
include_dir = os.path.join(
296+
conda_prefix, f'targets/{target_name}/include'
297+
)
295298
else:
296-
# A fallback when target cannot determined, though usually it shouldn't.
297-
include_dir = os.path.join(conda_prefix, f'include')
299+
# A fallback when target cannot determined
300+
# though usually it shouldn't.
301+
include_dir = os.path.join(conda_prefix, 'include')
298302
if os.path.exists(include_dir):
299303
return include_dir
300304
return None

0 commit comments

Comments
 (0)