From de7bf4867678491c76ffcc5b5054e9ab8c7e23df Mon Sep 17 00:00:00 2001
From: Patrick Lauer <patrick@gentoo.org>
Date: Sun, 9 Nov 2025 06:48:53 +0000
Subject: [PATCH] sci-libs/hipCUB: add 7.1.0

Signed-off-by: Patrick Lauer <patrick@gentoo.org>
---
 sci-libs/hipCUB/Manifest            |  1 +
 sci-libs/hipCUB/hipCUB-7.1.0.ebuild | 68 +++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+)
 create mode 100644 sci-libs/hipCUB/hipCUB-7.1.0.ebuild

diff --git a/sci-libs/hipCUB/Manifest b/sci-libs/hipCUB/Manifest
index d67762b99a714..095ee56c35b95 100644
--- a/sci-libs/hipCUB/Manifest
+++ b/sci-libs/hipCUB/Manifest
@@ -1,3 +1,4 @@
 DIST hipCUB-6.3.3.tar.gz 387580 BLAKE2B e0b605835be2b685c1cfa099a807e30fa061af8b14bafa1167c8249383230761a0d08532c831d3cb015e8e1b58119899d3cb7330990ca046045ea40058ee03e5 SHA512 2ec96088d34edeff48f8ee103a619f2ca7c741dcb020a3d1833a00e2072510790a563e98662aef7aa2a348034f3c8169ee39dcc61ab95202c734901ba53dfa4b
 DIST hipCUB-6.4.3.tar.gz 419173 BLAKE2B 7e1e707210191eee00aab01f8768a272f1fb5c35bb9e0635fe32ad50f4a8b56634d5f097f5eb2d1d5625f30f62450368a25a95ac2ba8fe85da2effd8efd669c9 SHA512 fb8018cf6200a468601149b6508b50944a27c3f7562012fef39e3a0ca13ed5ff597b1a3ad9e02273b70020ebd19ed10322e46d324f09fcf32a4f363e0ba72036
 DIST hipCUB-7.0.2.tar.gz 442494 BLAKE2B 62357955317e612f49023db12f844ea04ed8c16930000ce805c5a4bdd9c53207b6a064a5770bfb74f01f06872f72f0649b8322dda704816e608c2691ab6526cc SHA512 2c26a622c82e53d5e06cfb45066aac4bb4f7702b96d7753eb9acc0266fb8bc9c3f3068dba2acf023b1644be95bbe48e3a554db053e446757651615f9bd5ef7a6
+DIST hipCUB-7.1.0.tar.gz 471831 BLAKE2B edb368f37577600762a3c88d2af4c99e1696d033b66e6afb251e52b2bcc6baf8ed1375f75b69717421f449cada6a010eb652fceace5b408dbb3c6be004e8f7cc SHA512 165d08b75ef81becfce3c279f7ac8aaf0efee79d5d450450a8396b95f043b14716d8ac9c00047bec32ba21913b64ded51898e903959842fbf7d5a97328c7281a
diff --git a/sci-libs/hipCUB/hipCUB-7.1.0.ebuild b/sci-libs/hipCUB/hipCUB-7.1.0.ebuild
new file mode 100644
index 0000000000000..21704d0d8d57b
--- /dev/null
+++ b/sci-libs/hipCUB/hipCUB-7.1.0.ebuild
@@ -0,0 +1,68 @@
+# Copyright 1999-2025 Gentoo Authors
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=8
+
+ROCM_VERSION=${PV}
+
+inherit cmake rocm
+
+DESCRIPTION="Wrapper of rocPRIM or CUB for GPU parallel primitives"
+HOMEPAGE="https://github.com/ROCm/rocm-libraries/tree/develop/projects/hipcub"
+SRC_URI="https://github.com/ROCm/hipCUB/archive/rocm-${PV}.tar.gz -> hipCUB-${PV}.tar.gz"
+S="${WORKDIR}/hipCUB-rocm-${PV}"
+
+LICENSE="BSD"
+SLOT="0/$(ver_cut 1-2)"
+KEYWORDS="~amd64"
+IUSE="benchmark test"
+REQUIRED_USE="
+	benchmark? ( ${ROCM_REQUIRED_USE} )
+	test? ( ${ROCM_REQUIRED_USE} )
+"
+RESTRICT="!test? ( test )"
+
+RDEPEND="
+	benchmark? (
+		dev-util/hip:${SLOT}
+		dev-cpp/benchmark:=
+	)
+"
+DEPEND="
+	${RDEPEND}
+	dev-util/hip:${SLOT}
+	sci-libs/rocPRIM:${SLOT}
+	test? ( dev-cpp/gtest )
+"
+
+PATCHES=(
+	"${FILESDIR}"/${PN}-7.0.1-no-tests-install.patch
+)
+
+src_prepare() {
+	sed -e "s:set(ROCM_INSTALL_LIBDIR lib):set(ROCM_INSTALL_LIBDIR $(get_libdir)):" \
+		-i cmake/ROCMExportTargetsHeaderOnly.cmake || die
+
+	cmake_src_prepare
+}
+
+src_configure() {
+	rocm_use_clang
+
+	local mycmakeargs=(
+		-DGPU_TARGETS="$(get_amdgpu_flags)"
+		-DBUILD_TEST=$(usex test ON OFF)
+		-DBUILD_BENCHMARK=$(usex benchmark ON OFF)
+		-DBUILD_FILE_REORG_BACKWARD_COMPATIBILITY=OFF
+	)
+
+	cmake_src_configure
+}
+
+src_test() {
+	check_amdgpu
+	# Expected time on gfx1100 (-j32) is 85s
+	# HipcubDeviceHistogramMultiEven/0.MultiEven in 6.4.1 has bad array access (probably fixed in the future release)
+	local CMAKE_SKIP_TESTS=(hipcub.DeviceHistogram)
+	cmake_src_test
+}