Adding metadata

5 files changed, 92 insertions, 2 deletions

diff --git a/sci-ml/kineto/Manifest b/sci-ml/kineto/Manifest
index 0550d9811bea..90928ac4ee60 100644
--- a/sci-ml/kineto/Manifest
+++ b/sci-ml/kineto/Manifest
@@ -1,2 +1,3 @@
 DIST kineto-0.4.0_p20260323.tar.gz 8341701 BLAKE2B 2f1a3fdb5e9573f28460a5fe7b14a7bc40533fcf451e4f171a8bd1a46ca70db7469cf372ddb4ce344fbc0be1eb40d44055aff957e18c6e45a329daef34d45fe6 SHA512 65aa05794dca4ba29ddebd11845ebc3e218d46882bb2ccaa03a52566c2bc1696e4a151179d6ff93801576bc5d417e489fbd66c5e6dc2379b7e7fd2a4f5135e74
 DIST kineto-0.4.0_p20260603.tar.gz 257736 BLAKE2B a64298a936b8bd422c22b85084e663c84e5bba1be206332f946be97542950ef792ccfb13fcf486c475712d5ae74ad8a6afa8fc89a1dc0c9d0fa61470fe954133 SHA512 1f77eb52c17a4761f1f9772de196ba3400c5a9edacc2c3e558c2aafb685f3b4c7bf9402ffe5933cc5f87672e24329a730d5ad3522f7c9b13b0d1e15407509f14
+DIST kineto-0.4.0_p20260805.tar.gz 260453 BLAKE2B f8de006dfa9c2bae36a2319c59876279d96310a7a6e67c668f11f115098b766185df81b258ade23954b184c64236a8f6535b4322baae57c019f849448827d194 SHA512 9b57229f0930cdeb63eae3351a26918fc2f44f72c2090793a3ff0ef6201156110a0cef753b58679899edc151e18ffc6ae90e692fcc0694efc52b72c9c23224ed
diff --git a/sci-ml/kineto/files/kineto-0.4.0_p20260805-gentoo.patch b/sci-ml/kineto/files/kineto-0.4.0_p20260805-gentoo.patch
new file mode 100644
index 000000000000..51383a03a7d3
--- /dev/null
+++ b/sci-ml/kineto/files/kineto-0.4.0_p20260805-gentoo.patch
@@ -0,0 +1,29 @@
+--- a/CMakeLists.txt	2022-05-24 11:16:18.537610972 +0200
++++ b/CMakeLists.txt	2022-05-24 11:19:15.011073535 +0200
+@@ -209,10 +209,10 @@
+ # dynolog's ipcfabric is header-only and vendored under third_party/dynolog_headers
+ # (pinned in version.txt), so only its include paths are needed below -- there
+ # is no library to build or link.
+-set(DYNOLOG_INCLUDE_DIR "${LIBKINETO_THIRDPARTY_DIR}/dynolog_headers/")
+-set(IPCFABRIC_INCLUDE_DIR "${DYNOLOG_INCLUDE_DIR}/dynolog/src/ipcfabric/")
++set(DYNOLOG_INCLUDE_DIR "@GENTOO_PORTAGE_EPREFIX@/usr/include/dynolog/")
++set(IPCFABRIC_INCLUDE_DIR "@GENTOO_PORTAGE_EPREFIX@/usr/include/dynolog/src/ipcfabric/")
+ 
+-if(NOT TARGET fmt::fmt-header-only)
++if(FALSE)
+   if(NOT FMT_SOURCE_DIR)
+     set(FMT_SOURCE_DIR "${LIBKINETO_THIRDPARTY_DIR}/fmt"
+       CACHE STRING "fmt source directory from submodules")
+@@ -236,12 +236,10 @@
+ target_include_directories(kineto_base SYSTEM PUBLIC
+       $<BUILD_INTERFACE:${DYNOLOG_INCLUDE_DIR}>
+       $<BUILD_INTERFACE:${IPCFABRIC_INCLUDE_DIR}>)
+-target_link_libraries(kineto_base PRIVATE $<BUILD_INTERFACE:fmt::fmt-header-only>)
+ 
+ target_include_directories(kineto_api PUBLIC
+       $<BUILD_INTERFACE:${LIBKINETO_DIR}>
+       $<BUILD_INTERFACE:${LIBKINETO_INCLUDE_DIR}>)
+-target_link_libraries(kineto_api PRIVATE $<BUILD_INTERFACE:fmt::fmt-header-only>)
+ 
+ if(KINETO_LIBRARY_TYPE STREQUAL "static")
+   set(BUILD_SHARED_LIBS OFF)
diff --git a/sci-ml/kineto/kineto-0.4.0_p20260805.ebuild b/sci-ml/kineto/kineto-0.4.0_p20260805.ebuild
new file mode 100644
index 000000000000..57e92f2f175a
--- /dev/null
+++ b/sci-ml/kineto/kineto-0.4.0_p20260805.ebuild
@@ -0,0 +1,57 @@
+# Copyright 2022-2026 Gentoo Authors
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=8
+
+PYTHON_COMPAT=( python3_{13..14} )
+inherit python-any-r1 cmake prefix
+
+CommitId=15f4784a3471d4db8b31720d528f2f57cd2b1f40
+
+DESCRIPTION="part of the PyTorch Profiler"
+HOMEPAGE="https://github.com/pytorch/kineto"
+SRC_URI="https://github.com/pytorch/${PN}/archive/${CommitId}.tar.gz
+	-> ${P}.tar.gz"
+S="${WORKDIR}"/${PN}-${CommitId}/libkineto
+
+LICENSE="BSD"
+SLOT="0"
+KEYWORDS="~amd64 ~arm64"
+IUSE="rocm cuda test"
+
+RDEPEND="
+	dev-libs/libfmt
+	dev-libs/dynolog
+"
+DEPEND="${RDEPEND}"
+BDEPEND="
+	test? ( dev-cpp/gtest )
+	${PYTHON_DEPS}
+"
+RESTRICT="!test? ( test )"
+
+PATCHES=(
+	"${FILESDIR}"/${PN}-0.4.0_p20250418-AppxClock.patch
+)
+
+src_prepare() {
+	cmake_src_prepare
+}
+
+src_configure() {
+	local mycmakeargs=(
+		-DLIBKINETO_THIRDPARTY_DIR="${EPREFIX}"/usr/include/
+		-DKINETO_BUILD_TESTS=OFF # tests require cuda toolkit
+		-DCUDA_SOURCE_DIR=/opt/cuda
+	)
+	if use cuda; then
+		mycmakeargs+=( -DKINETO_BACKEND=cuda )
+	elif use rocm; then
+		mycmakeargs+=( -DKINETO_BACKEND=rocm )
+	else
+		mycmakeargs+=( -DKINETO_BACKEND=cpu )
+	fi
+	eapply $(prefixify_ro "${FILESDIR}"/${P}-gentoo.patch)
+
+	cmake_src_configure
+}
diff --git a/sci-ml/kineto/metadata.xml b/sci-ml/kineto/metadata.xml
index 31493193c995..b61df6e3e4af 100644
--- a/sci-ml/kineto/metadata.xml
+++ b/sci-ml/kineto/metadata.xml
@@ -5,5 +5,8 @@
     <email>tupone@gentoo.org</email>
     <name>Tupone Alfredo</name>
   </maintainer>
+  <use>
+    <flag name="rocm">Enable ROCm gpu computing support</flag>
+  </use>
   <origin>baldeagleos-repo</origin>
 </pkgmetadata>
diff --git a/sci-ml/pytorch/pytorch-2.14.0-r2.ebuild b/sci-ml/pytorch/pytorch-2.14.0-r3.ebuild
index 709b3d119b28..0e8558f1bd5c 100644
--- a/sci-ml/pytorch/pytorch-2.14.0-r2.ebuild
+++ b/sci-ml/pytorch/pytorch-2.14.0-r3.ebuild
@@ -89,7 +89,7 @@ RDEPEND="
 	)
 	fbgemm? ( >=sci-ml/FBGEMM-1.4 )
 	gloo? ( >=sci-ml/gloo-2025.06.04[cuda?,rocm?] )
-	kineto? ( ~sci-ml/kineto-0.4.0_p20260603 )
+	kineto? ( ~sci-ml/kineto-0.4.0_p20260805 )
 	mimalloc? ( dev-libs/mimalloc )
 	mkl? ( sci-libs/mkl )
 	mpi? ( virtual/mpi )
@@ -335,7 +335,7 @@ python_compile() {
 		addpredict "/dev/char/"
 
 		local -x CMAKE_CUDA_FLAGS="$(cuda_gccdir -f | tr -d \")"
-		local -x TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-7.0}"
+		local -x TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-7.5}"
 		local -x USE_CUDNN=ON
 		local -x USE_FLASH_ATTENTION=OFF
 		local -x USE_MEM_EFF_ATTENTION=OFF