Skip to content

Commit cebcc96

Browse files
committed
Set UMF_CUDA_INCLUDE_DIR to not fetch cudart from gitlab
Set UMF_CUDA_INCLUDE_DIR so that UMF does not fetch cudart from gitlab.com. Signed-off-by: Lukasz Dorau <[email protected]>
1 parent 16c8579 commit cebcc96

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

source/common/CMakeLists.txt

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,14 @@ else()
1111
set(UMF_BUILD_LEVEL_ZERO_PROVIDER OFF CACHE INTERNAL "Build Level Zero Provider")
1212
endif()
1313

14+
if (UR_BUILD_ADAPTER_CUDA)
15+
find_package(CUDA 10.1 REQUIRED)
16+
set(UMF_BUILD_CUDA_PROVIDER ON CACHE INTERNAL "Build UMF CUDA provider")
17+
set(UMF_CUDA_INCLUDE_DIR "${CUDA_INCLUDE_DIRS}" CACHE INTERNAL "CUDA headers")
18+
else()
19+
set(UMF_BUILD_CUDA_PROVIDER OFF CACHE INTERNAL "Build UMF CUDA provider")
20+
endif()
21+
1422
add_ur_library(ur_common STATIC
1523
ur_util.cpp
1624
ur_util.hpp
@@ -66,7 +74,6 @@ else()
6674
set(UMF_BUILD_EXAMPLES OFF CACHE INTERNAL "Build UMF examples")
6775
set(UMF_BUILD_SHARED_LIBRARY ${UMF_BUILD_SHARED_LIBRARY} CACHE INTERNAL "Build UMF shared library")
6876
set(UMF_BUILD_LIBUMF_POOL_DISJOINT ON CACHE INTERNAL "Build Disjoint Pool")
69-
set(UMF_BUILD_CUDA_PROVIDER ON CACHE INTERNAL "Build UMF CUDA provider")
7077

7178
FetchContent_MakeAvailable(unified-memory-framework)
7279
FetchContent_GetProperties(unified-memory-framework)

0 commit comments

Comments
 (0)