From 95e31e22a959e89e58073706492739c71d83c5e5 Mon Sep 17 00:00:00 2001 From: Chunmei Xu Date: Fri, 21 Jul 2023 14:02:18 +0800 Subject: [PATCH] enable cuda Signed-off-by: Chunmei Xu --- gloo.spec | 31 +++++++++++++++++++------------ 1 file changed, 19 insertions(+), 12 deletions(-) diff --git a/gloo.spec b/gloo.spec index 6f7ef71..f0111f6 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 -- Gitee