You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
4.8 KiB
137 lines
4.8 KiB
# Copyright 1999-2024 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=8
|
|
|
|
PYTHON_COMPAT=( python3_{10..11} )
|
|
ROCM_VERSION=${PV}
|
|
|
|
inherit cmake edo python-any-r1 toolchain-funcs rocm
|
|
|
|
DESCRIPTION="Basic Linear Algebra Subroutines for sparse computation"
|
|
HOMEPAGE="https://github.com/ROCmSoftwarePlatform/rocSPARSE"
|
|
|
|
SRC_URI="https://github.com/ROCmSoftwarePlatform/rocSPARSE/archive/rocm-${PV}.tar.gz -> rocSPARSE-${PV}.tar.gz
|
|
test? (
|
|
https://sparse.tamu.edu/MM/SNAP/amazon0312.tar.gz -> ${PN}_amazon0312.tar.gz
|
|
https://sparse.tamu.edu/MM/Muite/Chebyshev4.tar.gz -> ${PN}_Chebyshev4.tar.gz
|
|
https://sparse.tamu.edu/MM/FEMLAB/sme3Dc.tar.gz -> ${PN}_sme3Dc.tar.gz
|
|
https://sparse.tamu.edu/MM/Williams/webbase-1M.tar.gz -> ${PN}_webbase-1M.tar.gz
|
|
https://sparse.tamu.edu/MM/Bova/rma10.tar.gz -> ${PN}_rma10.tar.gz
|
|
https://sparse.tamu.edu/MM/JGD_BIBD/bibd_22_8.tar.gz -> ${PN}_bibd_22_8.tar.gz
|
|
https://sparse.tamu.edu/MM/Williams/mac_econ_fwd500.tar.gz -> ${PN}_mac_econ_fwd500.tar.gz
|
|
https://sparse.tamu.edu/MM/Williams/mc2depi.tar.gz -> ${PN}_mc2depi.tar.gz
|
|
https://sparse.tamu.edu/MM/Hamm/scircuit.tar.gz -> ${PN}_scircuit.tar.gz
|
|
https://sparse.tamu.edu/MM/Sandia/ASIC_320k.tar.gz -> ${PN}_ASIC_320k.tar.gz
|
|
https://sparse.tamu.edu/MM/GHS_psdef/bmwcra_1.tar.gz -> ${PN}_bmwcra_1.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos1.tar.gz -> ${PN}_nos1.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos2.tar.gz -> ${PN}_nos2.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos3.tar.gz -> ${PN}_nos3.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos4.tar.gz -> ${PN}_nos4.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos5.tar.gz -> ${PN}_nos5.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos6.tar.gz -> ${PN}_nos6.tar.gz
|
|
https://sparse.tamu.edu/MM/HB/nos7.tar.gz -> ${PN}_nos7.tar.gz
|
|
https://sparse.tamu.edu/MM/DNVS/shipsec1.tar.gz -> ${PN}_shipsec1.tar.gz
|
|
https://sparse.tamu.edu/MM/Cote/mplate.tar.gz -> ${PN}_mplate.tar.gz
|
|
https://sparse.tamu.edu/MM/Bai/qc2534.tar.gz -> ${PN}_qc2534.tar.gz
|
|
https://sparse.tamu.edu/MM/Chevron/Chevron2.tar.gz -> ${PN}_Chevron2.tar.gz
|
|
https://sparse.tamu.edu/MM/Chevron/Chevron3.tar.gz -> ${PN}_Chevron3.tar.gz
|
|
https://sparse.tamu.edu/MM/Chevron/Chevron4.tar.gz -> ${PN}_Chevron4.tar.gz
|
|
)"
|
|
|
|
LICENSE="MIT"
|
|
KEYWORDS="~amd64"
|
|
IUSE="benchmark test"
|
|
REQUIRED_USE="${ROCM_REQUIRED_USE}"
|
|
SLOT="0/$(ver_cut 1-2)"
|
|
|
|
RDEPEND="dev-util/hip
|
|
sci-libs/rocPRIM:${SLOT}"
|
|
DEPEND="${RDEPEND}"
|
|
BDEPEND="test? (
|
|
dev-cpp/gtest
|
|
>=dev-build/cmake-3.22
|
|
$(python_gen_any_dep 'dev-python/pyyaml[${PYTHON_USEDEP}]')
|
|
)
|
|
benchmark? ( app-admin/chrpath )
|
|
"
|
|
|
|
RESTRICT="!test? ( test )"
|
|
|
|
S="${WORKDIR}/rocSPARSE-rocm-${PV}"
|
|
|
|
PATCHES=( "${FILESDIR}/${PN}-5.0.2-remove-matrices-unpacking.patch"
|
|
"${FILESDIR}/${PN}-5.0.2-enable-gfx1031.patch"
|
|
"${FILESDIR}/${PN}-5.0.2-remove-incorrect-assert.patch" )
|
|
|
|
python_check_deps() {
|
|
if use test; then
|
|
python_has_version "dev-python/pyyaml[${PYTHON_USEDEP}]"
|
|
fi
|
|
}
|
|
|
|
src_prepare() {
|
|
sed -e "s/PREFIX rocsparse//" \
|
|
-e "/<INSTALL_INTERFACE/s,include,include/rocsparse," \
|
|
-e "/rocm_install_symlink_subdir(rocsparse)/d" \
|
|
-e "s:rocsparse/include:include/rocsparse:" \
|
|
-i "${S}/library/CMakeLists.txt" || die
|
|
|
|
# remove GIT dependency
|
|
sed -e "/find_package(Git/d" -i cmake/Dependencies.cmake || die
|
|
|
|
# Fix install path
|
|
sed -i -e "s.set(CMAKE_INSTALL_LIBDIR.#set(CMAKE_INSTALL_LIBDIR." CMakeLists.txt || die
|
|
|
|
# use python interpreter specifyied by python-any-r1
|
|
sed -e "/COMMAND ..\/common\/rocsparse_gentest.py/s,COMMAND ,COMMAND ${EPYTHON} ," -i clients/tests/CMakeLists.txt || die
|
|
|
|
cmake_src_prepare
|
|
|
|
# Test need download data from https://sparse.tamu.edu (or other mirror site), check MD5, unpack and convert them into csr format
|
|
# This process is handled default by ${S}/cmake/ClientMatrices.cmake, but should be the responsibility of portage.
|
|
if use test; then
|
|
mkdir -p "${BUILD_DIR}"/clients/matrices
|
|
# compile and use the mtx2csr converter. Do not use any optimization flags, because it causes error!
|
|
edo $(tc-getCXX) deps/convert.cpp -o deps/convert
|
|
find "${WORKDIR}" -maxdepth 2 -regextype grep -E -regex ".*/(.*)/\1\.mtx" -print0 |
|
|
while IFS= read -r -d '' mtxfile; do
|
|
destination=${BUILD_DIR}/clients/matrices/$(basename -s '.mtx' ${mtxfile}).csr
|
|
ebegin "Converting ${mtxfile} to ${destination}"
|
|
deps/convert ${mtxfile} ${destination}
|
|
eend $?
|
|
done
|
|
fi
|
|
}
|
|
|
|
src_configure() {
|
|
addpredict /dev/kfd
|
|
addpredict /dev/dri/
|
|
|
|
local mycmakeargs=(
|
|
-DCMAKE_SKIP_RPATH=On
|
|
-DAMDGPU_TARGETS="$(get_amdgpu_flags)"
|
|
-DBUILD_CLIENTS_SAMPLES=OFF
|
|
-DCMAKE_INSTALL_INCLUDEDIR="include/rocsparse"
|
|
-DBUILD_CLIENTS_TESTS=$(usex test ON OFF)
|
|
-DBUILD_CLIENTS_BENCHMARKS=$(usex benchmark ON OFF)
|
|
)
|
|
|
|
CXX=hipcc cmake_src_configure
|
|
}
|
|
|
|
src_test() {
|
|
check_amdgpu
|
|
cd "${BUILD_DIR}/clients/staging" || die
|
|
LD_LIBRARY_PATH="${BUILD_DIR}/library" edob ./${PN,,}-test
|
|
}
|
|
|
|
src_install() {
|
|
cmake_src_install
|
|
|
|
if use benchmark; then
|
|
cd "${BUILD_DIR}" || die
|
|
dobin clients/staging/rocsparse-bench
|
|
fi
|
|
}
|