diff options
Diffstat (limited to 'dev-libs/rccl/rccl-5.1.3.ebuild')
-rw-r--r-- | dev-libs/rccl/rccl-5.1.3.ebuild | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/dev-libs/rccl/rccl-5.1.3.ebuild b/dev-libs/rccl/rccl-5.1.3.ebuild new file mode 100644 index 000000000000..b7256c9d8ed4 --- /dev/null +++ b/dev-libs/rccl/rccl-5.1.3.ebuild @@ -0,0 +1,51 @@ +# Copyright 1999-2022 Gentoo Authors +# Distributed under the terms of the GNU General Public License v2 + +EAPI=8 + +ROCM_VERSION=${PV} + +inherit cmake edo rocm + +DESCRIPTION="ROCm Communication Collectives Library (RCCL)" +HOMEPAGE="https://github.com/ROCmSoftwarePlatform/rccl" +SRC_URI="https://github.com/ROCmSoftwarePlatform/rccl/archive/rocm-${PV}.tar.gz -> rccl-${PV}.tar.gz" + +LICENSE="BSD" +KEYWORDS="~amd64" +SLOT="0/$(ver_cut 1-2)" +IUSE="test" + +RDEPEND="dev-util/hip +dev-util/rocm-smi:${SLOT}" +DEPEND="${RDEPEND}" +BDEPEND=">=dev-util/cmake-3.22 + >=dev-util/rocm-cmake-5.0.2-r1 + test? ( dev-cpp/gtest )" + +RESTRICT="!test? ( test )" +S="${WORKDIR}/rccl-rocm-${PV}" + +PATCHES=( + "${FILESDIR}/${PN}-5.0.2-change_install_location.patch" + "${FILESDIR}/${PN}-5.1.3-remove-chrpath.patch" +) + +src_configure() { + addpredict /dev/kfd + addpredict /dev/dri/ + + local mycmakeargs=( + -DSKIP_RPATH=On + -DAMDGPU_TARGETS="$(get_amdgpu_flags)" + -DBUILD_TESTS=$(usex test ON OFF) + -Wno-dev + ) + + CXX=hipcc cmake_src_configure +} + +src_test() { + check_amdgpu + LD_LIBRARY_PATH="${BUILD_DIR}" edob test/UnitTests +} |