Skip to content

Commit 28e7206

Browse files
committed
Bug fix (existing code): conda cccl header directory
1 parent 6d490f5 commit 28e7206

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

cuda_pathfinder/cuda/pathfinder/_headers/find_nvidia_headers.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@ def _find_based_on_ctk_layout(libname: str, h_basename: str, anchor_point: str)
3838
parts.append("include")
3939
idir = os.path.join(*parts)
4040
if libname == "cccl":
41+
if IS_WINDOWS:
42+
cdir_ctk12 = os.path.join(idir, "targets", "x64")
43+
cdir_ctk13 = os.path.join(cdir_ctk12, "cccl")
44+
if _joined_isfile(cdir_ctk13, h_basename):
45+
return cdir_ctk13
46+
if _joined_isfile(cdir_ctk12, h_basename):
47+
return cdir_ctk12
48+
return None
4149
cdir = os.path.join(idir, "cccl") # CTK 13
4250
if _joined_isfile(cdir, h_basename):
4351
return cdir

0 commit comments

Comments
 (0)