We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 09d6ddd commit caf3d99Copy full SHA for caf3d99
1 file changed
cuda_core/tests/helpers.py
@@ -0,0 +1,15 @@
1
+# Copyright 2025 NVIDIA Corporation. All rights reserved.
2
+# SPDX-License-Identifier: Apache-2.0
3
+
4
+import os
5
6
+CUDA_PATH = os.environ.get("CUDA_PATH")
7
+CUDA_INCLUDE_PATH = None
8
+CCCL_INCLUDE_PATHS = None
9
+if CUDA_PATH is not None:
10
+ path = os.path.join(CUDA_PATH, "include")
11
+ if os.path.isdir(path):
12
+ CUDA_INCLUDE_PATH = path
13
+ path = os.path.join(path, "cccl")
14
15
+ CCCL_INCLUDE_PATHS = (path, CUDA_INCLUDE_PATH)
0 commit comments