mirror of
https://anongit.gentoo.org/git/repo/gentoo.git
synced 2025-12-16 07:01:08 +00:00
150 lines
3.6 KiB
Bash
150 lines
3.6 KiB
Bash
# Copyright 1999-2025 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=8
|
|
|
|
PYTHON_COMPAT=( python3_{10..13} )
|
|
ROCM_VERSION=${PV}
|
|
|
|
inherit cmake check-reqs edo multiprocessing python-r1 rocm
|
|
|
|
DESCRIPTION="Next generation FFT implementation for ROCm"
|
|
HOMEPAGE="https://github.com/ROCm/rocFFT"
|
|
SRC_URI="https://github.com/ROCm/rocFFT/archive/rocm-${PV}.tar.gz -> rocFFT-${PV}.tar.gz"
|
|
S="${WORKDIR}/rocFFT-rocm-${PV}"
|
|
|
|
LICENSE="MIT"
|
|
SLOT="0/$(ver_cut 1-2)"
|
|
KEYWORDS="~amd64"
|
|
|
|
# RDEPEND: perfscripts? dev-python/plotly[${PYTHON_USEDEP}] # currently masked by arch/amd64/x32/package.mask
|
|
RDEPEND="
|
|
perfscripts? (
|
|
>=media-gfx/asymptote-2.61
|
|
dev-texlive/texlive-latex
|
|
dev-tex/latexmk
|
|
sys-apps/texinfo
|
|
dev-python/sympy[${PYTHON_USEDEP}]
|
|
dev-python/numpy[${PYTHON_USEDEP}]
|
|
dev-python/scipy[${PYTHON_USEDEP}]
|
|
dev-python/pandas[${PYTHON_USEDEP}] )
|
|
${PYTHON_DEPS}"
|
|
|
|
DEPEND="=dev-util/hip-6*
|
|
${PYTHON_DEPS}
|
|
benchmark? (
|
|
dev-libs/boost
|
|
sci-libs/hipRAND:${SLOT}
|
|
)
|
|
test? (
|
|
dev-cpp/gtest
|
|
dev-libs/boost
|
|
>=sci-libs/fftw-3
|
|
llvm-runtimes/openmp
|
|
sci-libs/hipRAND:${SLOT}
|
|
)
|
|
"
|
|
|
|
BDEPEND="
|
|
>=dev-build/cmake-3.22
|
|
dev-build/rocm-cmake
|
|
dev-db/sqlite
|
|
"
|
|
|
|
CHECKREQS_DISK_BUILD="7G"
|
|
|
|
IUSE="benchmark perfscripts test"
|
|
REQUIRED_USE="perfscripts? ( benchmark ) ${PYTHON_REQUIRED_USE} ${ROCM_REQUIRED_USE}"
|
|
RESTRICT="!test? ( test )"
|
|
|
|
PATCHES=(
|
|
"${FILESDIR}"/${PN}-5.7.1-fix-rocm-link-path.patch
|
|
)
|
|
|
|
required_mem() {
|
|
if use test; then
|
|
echo "52G"
|
|
else
|
|
if [[ -n "${AMDGPU_TARGETS}" ]]; then
|
|
# count how many archs user specified in ${AMDGPU_TARGETS}
|
|
local NARCH=$(($(awk -F";" '{print NF-1}' <<< "${AMDGPU_TARGETS}" || die)+1))
|
|
else
|
|
# The default number of AMDGPU_TARGETS for rocFFT-4.3.0. May change in the future.
|
|
local NARCH=7
|
|
fi
|
|
echo "$(($(makeopts_jobs)*${NARCH}*25+2200))M" # A linear function estimating how much memory required
|
|
fi
|
|
}
|
|
|
|
pkg_pretend() {
|
|
return # leave the disk space check to pkg_setup phase
|
|
}
|
|
|
|
pkg_setup() {
|
|
export CHECKREQS_MEMORY=$(required_mem)
|
|
check-reqs_pkg_setup
|
|
python_setup
|
|
}
|
|
|
|
src_prepare() {
|
|
if use perfscripts; then
|
|
pushd scripts/perf || die
|
|
sed -e "/\/opt\/rocm/d" -e "/rocmversion/s,rocm_info.strip(),\"${PV}\"," -i perflib/specs.py || dir
|
|
sed -e "/^top/,+1d" -i rocfft-perf suites.py || die
|
|
sed -e "s,perflib,${PN}_perflib,g" -i rocfft-perf suites.py perflib/*.py || die
|
|
sed -e "/^top = /s,__file__).*$,\"${EPREFIX}/usr/share/${PN}-perflib\")," \
|
|
-i perflib/pdf.py perflib/generators.py || die
|
|
popd
|
|
fi
|
|
|
|
cmake_src_prepare
|
|
}
|
|
|
|
src_configure() {
|
|
rocm_use_hipcc
|
|
|
|
local mycmakeargs=(
|
|
-DCMAKE_SKIP_RPATH=ON
|
|
-DGPU_TARGETS="$(get_amdgpu_flags)"
|
|
-Wno-dev
|
|
-DROCM_SYMLINK_LIBS=OFF
|
|
-DBUILD_CLIENTS_TESTS=$(usex test ON OFF)
|
|
-DBUILD_CLIENTS_BENCH=$(usex benchmark ON OFF)
|
|
-DSQLITE_USE_SYSTEM_PACKAGE=ON
|
|
-DBUILD_FILE_REORG_BACKWARD_COMPATIBILITY=OFF
|
|
)
|
|
|
|
cmake_src_configure
|
|
}
|
|
|
|
src_test() {
|
|
check_amdgpu
|
|
cd "${BUILD_DIR}/clients/staging" || die
|
|
export LD_LIBRARY_PATH=${BUILD_DIR}/library/src/:${BUILD_DIR}/library/src/device
|
|
edob ./rocfft-test
|
|
}
|
|
|
|
src_install() {
|
|
cmake_src_install
|
|
|
|
if use benchmark; then
|
|
cd "${BUILD_DIR}"/clients/staging || die
|
|
dobin dyna-rocfft-bench rocfft-bench
|
|
dosym dyna-rocfft-bench /usr/bin/dyna-rocfft-rider
|
|
dosym rocfft-bench /usr/bin/dyna-rocfft-rider
|
|
|
|
if ! use perfscripts; then
|
|
# prevent collision with dev-util/perf
|
|
rm -rf "${ED}"/usr/bin/perf || die
|
|
fi
|
|
fi
|
|
|
|
if use perfscripts; then
|
|
cd "${S}"/scripts/perf || die
|
|
python_foreach_impl python_doexe rocfft-perf
|
|
python_moduleinto ${PN}_perflib
|
|
python_foreach_impl python_domodule perflib/*.py
|
|
insinto /usr/share/${PN}-perflib
|
|
doins *.asy suites.py
|
|
fi
|
|
}
|