File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2828
2929try :
3030 from cuda .pathfinder ._utils .env_vars import get_cuda_home_or_path
31+
3132 CUDA_HOME = get_cuda_home_or_path ()
3233except ImportError :
3334 # Fallback for build environments where cuda-pathfinder may not be available
Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments