[gentoo-commits] repo/gentoo:master commit in: sci-ml/accelerate/

[Alfredo Tupone]

commit:     528ba848b9541e01464fc2d7c94bd64becf7389c
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Thu Apr 17 18:00:04 2025 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Thu Apr 17 18:00:19 2025 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=528ba848

sci-ml/accelerate: add 1.6.0

Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-ml/accelerate/Manifest                |  1 +
 sci-ml/accelerate/accelerate-1.6.0.ebuild | 53 +++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)

diff --git a/sci-ml/accelerate/Manifest b/sci-ml/accelerate/Manifest
index 94987c8942c1..4d6a90c625bd 100644
--- a/sci-ml/accelerate/Manifest
+++ b/sci-ml/accelerate/Manifest
@@ -1 +1,2 @@
 DIST accelerate-1.5.2.gh.tar.gz 765155 BLAKE2B 
4ae488772251b0f00ff4e195001299fbc22dd12b9625547fbfb952b92a2c4c2a2ea9e1ac976e25b6a1d74ce4bfca4d0398362dfe30cbb266f297539ce790fc1c
 SHA512 
aebdc3963fbc4d331bc8ed37b8769d13ec09cfe1f6b1607c83e2cc92903ef4e5ab6d3be5eb6e1007f3efa1a06e80e7a2d152e3180f2613baebcf1f787ac53139
+DIST accelerate-1.6.0.gh.tar.gz 963945 BLAKE2B 
2d3c006f8290414279a53b4bea725e8227ff69288bb4984bfbef00fa0f40999fcd53faf1ff45e351e2c36020b448970e7a36415e6de0cac3c6952bea17e5e62b
 SHA512 
cb516be77c9c05313d6a79fc846a2635fd3635e0391d061deb22692eb7b8ff9bcb2fa03926ccf76587f6d929dcd1a9089cdb2d2004567befc5f4daca71c05224

diff --git a/sci-ml/accelerate/accelerate-1.6.0.ebuild 
b/sci-ml/accelerate/accelerate-1.6.0.ebuild
new file mode 100644
index 000000000000..3ea32e16b7f2
--- /dev/null
+++ b/sci-ml/accelerate/accelerate-1.6.0.ebuild
@@ -0,0 +1,53 @@
+# Copyright 2025 Gentoo Authors
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=8
+PYTHON_COMPAT=( python3_{10..12} )
+DISTUTILS_USE_PEP517=setuptools
+DISTUTILS_SINGLE_IMPL=1
+inherit distutils-r1
+
+DESCRIPTION="Run your *raw* PyTorch training script on any kind of device"
+HOMEPAGE="https://github.com/huggingface/accelerate";
+SRC_URI="https://github.com/huggingface/${PN}/archive/refs/tags/v${PV}.tar.gz
+       -> ${P}.gh.tar.gz"
+
+LICENSE="Apache-2.0"
+SLOT="0"
+KEYWORDS="~amd64"
+
+RDEPEND="
+       sci-ml/huggingface_hub[${PYTHON_SINGLE_USEDEP}]
+       sci-ml/pytorch[${PYTHON_SINGLE_USEDEP}]
+       $(python_gen_cond_dep '
+               dev-python/numpy[${PYTHON_USEDEP}]
+               dev-python/packaging[${PYTHON_USEDEP}]
+               dev-python/psutil[${PYTHON_USEDEP}]
+               dev-python/pyyaml[${PYTHON_USEDEP}]
+               sci-ml/safetensors[${PYTHON_USEDEP}]
+       ')
+"
+DEPEND="${RDEPEND}"
+BDEPEND="test? (
+       sci-ml/caffe2[gloo]
+)"
+
+distutils_enable_tests pytest
+
+python_test() {
+       local EPYTEST_IGNORE=(
+               tests/test_examples.py
+               tests/deepspeed
+               tests/test_big_modeling.py
+               tests/fsdp
+               tests/tp
+               tests/test_cli.py
+       )
+       local EPYTEST_DESELECT=(
+               
tests/test_modeling_utils.py::ModelingUtilsTester::test_infer_auto_device_map_with_buffer_check
+               
tests/test_modeling_utils.py::ModelingUtilsTester::test_infer_auto_device_map_with_buffer_check_and_multi_devices
+               
tests/test_modeling_utils.py::ModelingUtilsTester::test_infer_auto_device_map_with_fallback_allocation_and_buffers
+               
tests/test_utils.py::UtilsTester::test_patch_environment_key_exists
+       )
+       epytest tests
+}