blob: bf7e339f002a70ad3f3f725f08b3bd540b3013d3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# Copyright 1999-2022 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
inherit cmake
DESCRIPTION="CU / ROCM agnostic hip FFT implementation"
HOMEPAGE="https://github.com/ROCmSoftwarePlatform/hipFFT"
SRC_URI="https://github.com/ROCmSoftwarePlatform/hipFFT/archive/refs/tags/rocm-${PV}.tar.gz -> hipFFT-rocm-${PV}.tar.gz
test? ( https://github.com/ROCmSoftwarePlatform/rocFFT/archive/rocm-${PV}.tar.gz -> rocFFT-${PV}.tar.gz )"
LICENSE="MIT"
KEYWORDS="~amd64"
IUSE="benchmark test"
SLOT="0/$(ver_cut 1-2)"
RESTRICT="!test? ( test )"
RDEPEND="dev-util/hip:${SLOT}
sci-libs/rocFFT:${SLOT}"
DEPEND="${RDEPEND}"
BDEPEND="
test? (
dev-cpp/gtest
dev-libs/boost
)"
S="${WORKDIR}/hipFFT-rocm-${PV}"
PATCHES=(
"${FILESDIR}/${PN}-5.0.2-gentoo-install-locations.patch"
"${FILESDIR}/${PN}-5.0.2-remove-git-dependency.patch"
"${FILESDIR}/${PN}-4.3.0-add-complex-header.patch"
)
src_prepare() {
use test && rmdir rocFFT && ln -s ../rocFFT-rocm-${PV} rocFFT
sed -e "/CMAKE_INSTALL_LIBDIR/d" -i CMakeLists.txt || die
CXX=hipcc cmake_src_prepare
}
src_configure() {
# Grant access to the device
addwrite /dev/kfd
addpredict /dev/dri/
local mycmakeargs=(
-DCMAKE_INSTALL_PREFIX="${EPREFIX}/usr"
-DCMAKE_INSTALL_INCLUDEDIR="include/hipfft"
-DCMAKE_SKIP_RPATH=ON
-DCMAKE_MODULE_PATH="${EPREFIX}/usr/$(get_libdir)/cmake"
-DHIP_ROOT_DIR="${EPREFIX}/usr"
-DBUILD_CLIENTS_TESTS=$(usex test ON OFF)
-DBUILD_CLIENTS_RIDER=$(usex benchmark ON OFF)
${AMDGPU_TARGETS+-DAMDGPU_TARGETS="${AMDGPU_TARGETS}"}
)
export CXX=hipcc
cmake_src_configure
}
src_test () {
addwrite /dev/kfd
addpredict /dev/dri
cd "${BUILD_DIR}/clients/staging" || die
einfo "Running hipfft-test"
LD_LIBRARY_PATH=${BUILD_DIR}/library ./hipfft-test || die
}
src_install() {
cmake_src_install
if use benchmark; then
cd "${BUILD_DIR}/clients/staging" || die
dobin hipfft-rider
fi
}
|