Skip to content

Commit

Permalink
Added CUDA platforms for x86_64-linux-gnu
Browse files Browse the repository at this point in the history
  • Loading branch information
stemann committed Mar 7, 2022
1 parent 697d653 commit 4292d5a
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions G/Gloo/build_tarballs.jl
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,16 @@ script = raw"""
cd $WORKSPACE/srcdir/gloo
mkdir build
cd build
if [[ $bb_full_target == *cuda* ]]; then
cmake_extra_args="-DUSE_CUDA=ON -DCUDA_TOOLKIT_ROOT_DIR=$prefix/cuda"
fi
cmake \
-DCMAKE_INSTALL_PREFIX=$prefix \
-DCMAKE_TOOLCHAIN_FILE=${CMAKE_TARGET_TOOLCHAIN} \
-DCMAKE_BUILD_TYPE=Release \
-DBUILD_SHARED_LIBS=ON \
-DUSE_LIBUV=ON \
$cmake_extra_args \
..
cmake --build . -- -j $nproc
make install
Expand All @@ -31,7 +35,14 @@ make install
platforms = supported_platforms()
filter!(p -> nbits(p) == 64, platforms) # Gloo can only be built on 64-bit systems
filter!(!Sys.iswindows, platforms) # Windows support will be available from 20200910, i.e. 881f7f0dcf06f7e49e134a45d3284860fb244fa9
cuda_platforms = Platform[]
for cuda_version in [v"10.2", v"11.0", v"11.1", v"11.2", v"11.3", v"11.4", v"11.5", v"11.6"]
cuda_platform = Platform("x86_64", "linux"; cuda = "$(cuda_version.major).$(cuda_version.minor)")
push!(platforms, cuda_platform)
push!(cuda_platforms, cuda_platform)
end
platforms = expand_cxxstring_abis(platforms)
cuda_platforms = expand_cxxstring_abis(cuda_platforms)

# The products that we will ensure are always built
products = [
Expand All @@ -41,6 +52,7 @@ products = [
# Dependencies that must be installed before this package can be built
dependencies = [
Dependency("LibUV_jll", v"2.0.0"),
BuildDependency("CUDA_full_jll"; platforms = cuda_platforms),
]

# Build the tarballs, and possibly a `build.jl` as well.
Expand Down

0 comments on commit 4292d5a

Please sign in to comment.