Skip to content

Commit 44016f2

Browse files
committed
a
1 parent 0c44ef6 commit 44016f2

File tree

3 files changed

+91
-18
lines changed

3 files changed

+91
-18
lines changed

.github/workflows/reusable_basic.yml

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ jobs:
2626
install_tbb: ['ON']
2727
disable_hwloc: ['OFF']
2828
link_hwloc_statically: ['OFF']
29+
cmake_ver: ['latest']
2930
include:
3031
- os: 'ubuntu-22.04'
3132
build_type: Release
@@ -36,6 +37,8 @@ jobs:
3637
install_tbb: 'ON'
3738
disable_hwloc: 'OFF'
3839
link_hwloc_statically: 'OFF'
40+
# check minimum supported cmake version
41+
cmake_ver: '3.14.0'
3942
- os: 'ubuntu-22.04'
4043
build_type: Release
4144
compiler: {c: gcc, cxx: g++}
@@ -45,6 +48,7 @@ jobs:
4548
install_tbb: 'ON'
4649
disable_hwloc: 'OFF'
4750
link_hwloc_statically: 'OFF'
51+
cmake_ver: 'latest'
4852
- os: 'ubuntu-24.04'
4953
build_type: Debug
5054
compiler: {c: gcc, cxx: g++}
@@ -54,6 +58,7 @@ jobs:
5458
install_tbb: 'ON'
5559
disable_hwloc: 'OFF'
5660
link_hwloc_statically: 'OFF'
61+
cmake_ver: 'latest'
5762
# test level_zero_provider='OFF' and cuda_provider='OFF'
5863
- os: 'ubuntu-22.04'
5964
build_type: Release
@@ -64,6 +69,7 @@ jobs:
6469
install_tbb: 'ON'
6570
disable_hwloc: 'OFF'
6671
link_hwloc_statically: 'OFF'
72+
cmake_ver: 'latest'
6773
# test icx compiler
6874
- os: 'ubuntu-22.04'
6975
build_type: Release
@@ -74,6 +80,7 @@ jobs:
7480
install_tbb: 'ON'
7581
disable_hwloc: 'OFF'
7682
link_hwloc_statically: 'OFF'
83+
cmake_ver: 'latest'
7784
# test lld linker
7885
- os: 'ubuntu-24.04'
7986
build_type: Release
@@ -85,7 +92,8 @@ jobs:
8592
disable_hwloc: 'OFF'
8693
link_hwloc_statically: 'OFF'
8794
llvm_linker: '-DCMAKE_EXE_LINKER_FLAGS="-fuse-ld=lld" -DCMAKE_MODULE_LINKER_FLAGS="-fuse-ld=lld" -DCMAKE_SHARED_LINKER_FLAGS="-fuse-ld=lld"'
88-
# test without installing TBB
95+
cmake_ver: 'latest'
96+
# test without installing TBB
8997
- os: 'ubuntu-22.04'
9098
build_type: Release
9199
compiler: {c: gcc, cxx: g++}
@@ -95,6 +103,7 @@ jobs:
95103
install_tbb: 'OFF'
96104
disable_hwloc: 'OFF'
97105
link_hwloc_statically: 'OFF'
106+
cmake_ver: 'latest'
98107
- os: 'ubuntu-22.04'
99108
build_type: Debug
100109
compiler: {c: gcc, cxx: g++}
@@ -104,6 +113,7 @@ jobs:
104113
install_tbb: 'ON'
105114
disable_hwloc: 'ON'
106115
link_hwloc_statically: 'OFF'
116+
cmake_ver: 'latest'
107117
- os: 'ubuntu-22.04'
108118
build_type: Release
109119
compiler: {c: gcc, cxx: g++}
@@ -113,6 +123,7 @@ jobs:
113123
install_tbb: 'ON'
114124
disable_hwloc: 'OFF'
115125
link_hwloc_statically: 'ON'
126+
cmake_ver: 'latest'
116127
runs-on: ${{matrix.os}}
117128

118129
steps:
@@ -124,7 +135,19 @@ jobs:
124135
- name: Install apt packages
125136
run: |
126137
sudo apt-get update
127-
sudo apt-get install -y clang cmake libnuma-dev lcov
138+
sudo apt-get install -y clang libnuma-dev lcov
139+
140+
- name: Install cmake (latest)
141+
if: matrix.cmake_ver == 'latest'
142+
run: |
143+
sudo apt-get install -y cmake
144+
145+
- name: Install cmake (minimum supported version)
146+
if: matrix.cmake_ver != 'latest'
147+
run: |
148+
wget https://github.com/Kitware/CMake/releases/download/v${{matrix.cmake_ver}}/cmake-${{matrix.cmake_ver}}-Linux-x86_64.sh
149+
chmod +x cmake-${{matrix.cmake_ver}}-Linux-x86_64.sh
150+
sudo ./cmake-${{matrix.cmake_ver}}-Linux-x86_64.sh --skip-license --prefix=/usr/local
128151
129152
- name: Install hwloc
130153
if: matrix.disable_hwloc == 'OFF'

CMakeLists.txt

Lines changed: 34 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,9 @@
22
# Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
33
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
44

5+
message(STATUS "CMake version: ${CMAKE_VERSION}")
56
cmake_minimum_required(VERSION 3.14.0 FATAL_ERROR)
7+
68
# needed when UMF is used as an external project
79
set(UMF_CMAKE_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR})
810

@@ -404,16 +406,17 @@ if(UMF_BUILD_LEVEL_ZERO_PROVIDER AND (NOT UMF_LEVEL_ZERO_INCLUDE_DIR))
404406
GIT_REPOSITORY ${LEVEL_ZERO_LOADER_REPO}
405407
GIT_TAG ${LEVEL_ZERO_LOADER_TAG}
406408
EXCLUDE_FROM_ALL)
407-
FetchContent_MakeAvailable(level-zero-loader)
409+
# Only populate the repo - we don't need to build it
410+
FetchContent_Populate(level-zero-loader)
408411

409412
set(LEVEL_ZERO_INCLUDE_DIRS
410413
${level-zero-loader_SOURCE_DIR}/include
411-
CACHE PATH "Path to Level Zero Headers")
412-
message(STATUS "Level Zero include directory: ${LEVEL_ZERO_INCLUDE_DIRS}")
414+
CACHE PATH "Path to Level Zero headers")
415+
message(STATUS "LEVEL_ZERO_INCLUDE_DIRS = ${LEVEL_ZERO_INCLUDE_DIRS}")
413416
elseif(UMF_BUILD_LEVEL_ZERO_PROVIDER)
414417
# Only header is needed to build UMF
415418
set(LEVEL_ZERO_INCLUDE_DIRS ${UMF_LEVEL_ZERO_INCLUDE_DIR})
416-
message(STATUS "Level Zero include directory: ${LEVEL_ZERO_INCLUDE_DIRS}")
419+
message(STATUS "LEVEL_ZERO_INCLUDE_DIRS = ${LEVEL_ZERO_INCLUDE_DIRS}")
417420
endif()
418421

419422
# Fetch CUDA only if needed i.e.: if building CUDA provider is ON and CUDA
@@ -423,14 +426,15 @@ if(UMF_BUILD_CUDA_PROVIDER AND (NOT UMF_CUDA_INCLUDE_DIR))
423426
"https://gitlab.com/nvidia/headers/cuda-individual/cudart.git")
424427
set(CUDA_TAG cuda-12.5.1)
425428

426-
message(STATUS "Fetching CUDA ${CUDA_TAG} from ${CUDA_REPO} ...")
429+
message(STATUS "Fetching CUDA (${CUDA_TAG}) from ${CUDA_REPO} ...")
427430

428431
FetchContent_Declare(
429432
cuda-headers
430433
GIT_REPOSITORY ${CUDA_REPO}
431434
GIT_TAG ${CUDA_TAG}
432435
EXCLUDE_FROM_ALL)
433-
FetchContent_MakeAvailable(cuda-headers)
436+
# Only populate the repo - we don't need to build it
437+
FetchContent_Populate(cuda-headers)
434438

435439
set(CUDA_INCLUDE_DIRS
436440
${cuda-headers_SOURCE_DIR}
@@ -707,16 +711,31 @@ else()
707711
)
708712
endif()
709713

710-
if((UMF_BUILD_GPU_TESTS OR UMF_BUILD_GPU_EXAMPLES) AND UMF_BUILD_CUDA_PROVIDER)
711-
find_package(CUDA REQUIRED cuda)
712-
if(CUDA_LIBRARIES)
713-
set(UMF_CUDA_ENABLED TRUE)
714-
else()
715-
message(
716-
STATUS "Disabling tests and examples that use the CUDA provider "
717-
"because the CUDA libraries they require were not found.")
714+
if(UMF_BUILD_GPU_TESTS OR UMF_BUILD_GPU_EXAMPLES)
715+
if(UMF_BUILD_CUDA_PROVIDER)
716+
find_package(CUDA REQUIRED cuda)
717+
if(CUDA_LIBRARIES)
718+
set(UMF_CUDA_ENABLED TRUE)
719+
else()
720+
message(
721+
STATUS
722+
"Disabling tests and examples that use the CUDA Provider "
723+
"because the CUDA libraries they require were not found.")
724+
endif()
725+
endif()
726+
727+
if(UMF_BUILD_LEVEL_ZERO_PROVIDER)
728+
find_package(LEVEL_ZERO REQUIRED level-zero)
729+
if(LEVEL_ZERO_LIBRARIES)
730+
set(UMF_LEVEL_ZERO_ENABLED TRUE)
731+
else()
732+
message(
733+
STATUS
734+
"Disabling tests and examples that use the Level Zero Provider "
735+
"because the Level Zero libraries they require were not found."
736+
)
737+
endif()
718738
endif()
719-
# TODO do the same for ze_loader
720739
endif()
721740

722741
add_subdirectory(src)

test/test_installation.py

Lines changed: 32 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import subprocess # nosec B404
1414
import sys
1515
from typing import List
16+
import os
1617

1718

1819
class UmfInstaller:
@@ -57,6 +58,23 @@ def __init__(
5758
self.hwloc = hwloc
5859
self.match_list = self._create_match_list()
5960

61+
def _get_cmake_major_minor_version(self) -> List[int]:
62+
"""
63+
Returns the version of CMake
64+
"""
65+
66+
cmake_cmd = "cmake --version"
67+
try:
68+
output = subprocess.check_output(cmake_cmd.split()).decode() # nosec B603
69+
version_line = output.splitlines()[0]
70+
cmake_version = version_line.split()[2]
71+
# get major and minor version
72+
major_minor = [int(x) for x in cmake_version.split(".")[:2]]
73+
print("CMake version:", major_minor, flush=True)
74+
return major_minor
75+
except subprocess.CalledProcessError:
76+
sys.exit(f"Error: Unable to get CMake version using command '{cmake_cmd}'")
77+
6078
def _create_match_list(self) -> List[str]:
6179
"""
6280
Defines a list of relative paths to files that should be installed.
@@ -182,8 +200,21 @@ def install_umf(self) -> None:
182200
)
183201

184202
install_cmd = f"cmake --install {self.build_dir} --config {self.build_type.title()} --prefix {self.install_dir}"
203+
env = os.environ.copy()
204+
205+
# use make to install UMF if CMake version is lower than 3.15
206+
major, minor = self._get_cmake_major_minor_version()
207+
if major < 3 and minor < 15:
208+
install_cmd = "make -C " + str(self.build_dir) + " install"
209+
env["CMAKE_INSTALL_PREFIX"] = str(self.install_dir)
210+
env["CMAKE_BUILD_TYPE"] = self.build_type.title()
211+
185212
try:
186-
subprocess.run(install_cmd.split()).check_returncode() # nosec B603
213+
print(f"Running command: {install_cmd}", flush=True)
214+
subprocess.run(
215+
install_cmd.split(), env=env
216+
).check_returncode() # nosec B603
217+
187218
except subprocess.CalledProcessError:
188219
sys.exit(f"Error: UMF installation command '{install_cmd}' failed")
189220

0 commit comments

Comments
 (0)