diff --git a/gloo.spec b/gloo.spec index 6f7ef714b4c1b8bc327445ced10a5356c016e253..f0111f67d04db4af0f3f4225e5947de1f0c14010 100644 --- a/gloo.spec +++ b/gloo.spec @@ -1,9 +1,17 @@ +%define anolis_release 1 + %global githash c6f3a5b %global gitdate 202306162 +%global have_cuda 1 +%global cu_maj 12 +%global cu_min 1 +%global __cmake_in_source_build 1 +%global gpu_target_bin "5.2;6.1;7.5;8.6" +%global gpu_target_ptx "5.2" Name: gloo Version: 0 -Release: %{gitdate}.git%{githash}%{?dist} +Release: %{gitdate}.git%{githash}.%{anolis_release}%{?dist} Summary: Collective communications library License: BSD URL: https://github.com/facebookincubator/gloo @@ -12,18 +20,10 @@ Source0: %{name}.tar.xz BuildRequires: cmake gcc-c++ hiredis-devel libuv-devel rdma-core-devel -%global have_cuda 0 -%global __cmake_in_source_build 1 - -%bcond_with cuda -%if %{without cuda} -%global have_cuda 0 -%endif - %if %{have_cuda} -BuildRequires: cuda-nvcc-11-4 -BuildRequires: cuda-cudart-devel-11-4 -BuildRequires: cuda-nvml-devel-11-4 +BuildRequires: cuda-nvcc-%{cu_maj}-%{cu_min} +BuildRequires: cuda-cudart-devel-%{cu_maj}-%{cu_min} +BuildRequires: cuda-nvml-devel-%{cu_maj}-%{cu_min} BuildRequires: libnccl-devel Requires: libnccl %endif @@ -77,7 +77,11 @@ pushd build -DUSE_CUDA=ON \ -DUSE_NCCL=ON \ -DNVCC_FLAGS_EXTRA="Auto" \ + -DCUDA_GENERATION="" \ + -DCUDA_ARCH_BIN=%{gpu_target_bin} \ + -DCUDA_ARCH_PTX=%{gpu_target_ptx} \ -DCUDA_PROPAGATE_HOST_FLAGS=OFF \ + -DCUDA_NVCC_FLAGS="-Wno-deprecated-gpu-targets -allow-unsupported-compiler" \ %else -DUSE_CUDA=OFF \ -DUSE_NCCL=OFF \ @@ -114,5 +118,8 @@ popd %changelog +* Fri Jul 21 2023 Chunmei Xu - 0-202306162.gitc6f3a5b.1 +- enable cuda + * Fri Jun 16 2023 forrest_ly - 0-202306162.gitc6f3a5b - init for anolis 23