Skip to content

Commit 741f6c6

Browse files
committed
wip
1 parent b127a47 commit 741f6c6

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

cuda_bindings/setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828

2929
try:
3030
from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
31+
3132
CUDA_HOME = get_cuda_home_or_path()
3233
except ImportError:
3334
# Fallback for build environments where cuda-pathfinder may not be available

cuda_core/build_hooks.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,12 @@ def _get_cuda_paths() -> list[str]:
3333
"""
3434
try:
3535
from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
36+
3637
CUDA_PATH = get_cuda_home_or_path()
3738
except ImportError:
3839
# Fallback for build environments where cuda-pathfinder may not be available
3940
CUDA_PATH = os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME", None))
40-
41+
4142
if not CUDA_PATH:
4243
raise RuntimeError("Environment variable CUDA_PATH or CUDA_HOME is not set")
4344
CUDA_PATH = CUDA_PATH.split(os.pathsep)

0 commit comments

Comments
 (0)