summaryrefslogtreecommitdiff
path: root/dev-libs/cutlass/cutlass-2.10.0-r1.ebuild
diff options
context:
space:
mode:
Diffstat (limited to 'dev-libs/cutlass/cutlass-2.10.0-r1.ebuild')
-rw-r--r--dev-libs/cutlass/cutlass-2.10.0-r1.ebuild39
1 files changed, 39 insertions, 0 deletions
diff --git a/dev-libs/cutlass/cutlass-2.10.0-r1.ebuild b/dev-libs/cutlass/cutlass-2.10.0-r1.ebuild
new file mode 100644
index 000000000000..4ce3100f8132
--- /dev/null
+++ b/dev-libs/cutlass/cutlass-2.10.0-r1.ebuild
@@ -0,0 +1,39 @@
+# Copyright 2023 Gentoo Authors
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=8
+
+inherit cuda cmake
+
+DESCRIPTION="CUDA Templates for Linear Algebra Subroutines"
+HOMEPAGE="https://github.com/NVIDIA/cutlass"
+SRC_URI="https://github.com/NVIDIA/${PN}/archive/refs/tags/v${PV}.tar.gz
+ -> ${P}.tar.gz"
+
+LICENSE="BSD"
+SLOT="0"
+KEYWORDS="~amd64"
+
+DEPEND=""
+RDEPEND="${DEPEND}"
+BDEPEND="dev-util/nvidia-cuda-toolkit"
+
+src_prepare() {
+ cmake_src_prepare
+ cuda_src_prepare
+}
+
+src_configure() {
+ mycmakeargs+=(
+ -DCMAKE_CUDA_FLAGS="$(cuda_gccdir -f | tr -d \")"
+ -DCUTLASS_ENABLE_HEADERS_ONLY=yes
+ -DCUTLASS_ENABLE_TESTS=no
+ )
+ cuda_add_sandbox -w
+ cmake_src_configure
+}
+
+src_install() {
+ cmake_src_install
+ rm -r "${D}"/usr/test || die
+}