@@ -262,7 +262,7 @@ def get_debian_pkg_libdevice():
262262
263263def 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+
284285def 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