summaryrefslogtreecommitdiff
path: root/sci-ml/accelerate
diff options
context:
space:
mode:
authorV3n3RiX <venerix@koprulu.sector>2025-04-18 08:46:05 +0100
committerV3n3RiX <venerix@koprulu.sector>2025-04-18 08:46:05 +0100
commitcf1a411e806b434c96329401a6ff0901ec7f0d5c (patch)
tree969527fcceb3b686400c73b5ed22c1d587a8eb3e /sci-ml/accelerate
parent6a01345ec5dc673d0ebd4653ceb063eafa7fcf65 (diff)
gentoo auto-resync : 18:04:2025 - 08:46:04next
Diffstat (limited to 'sci-ml/accelerate')
-rw-r--r--sci-ml/accelerate/Manifest2
-rw-r--r--sci-ml/accelerate/accelerate-1.6.0.ebuild53
2 files changed, 55 insertions, 0 deletions
diff --git a/sci-ml/accelerate/Manifest b/sci-ml/accelerate/Manifest
index b259de850602..e0324aee4cf0 100644
--- a/sci-ml/accelerate/Manifest
+++ b/sci-ml/accelerate/Manifest
@@ -1,3 +1,5 @@
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
EBUILD accelerate-1.5.2-r2.ebuild 1248 BLAKE2B 7c9b6def3abd904edf64ebe3bf3297df4e4a7f9b5e44eab82181e59bcd0fce66b9cf7d652bf940c7ddecf966d18cdf67a85e8a651a8754da235737bf7b53b76e SHA512 493829281c7102226e0026d84d0fe7fcc6688653b324b272ff15fa23043d598fbfbe51518e04b8a8945b0cd96b375150e1d4c622f2ad1ed03e2e1434f3648a4a
+EBUILD accelerate-1.6.0.ebuild 1519 BLAKE2B ded228d6cac06bc7e54e6fedb8c24248dc20dedbf7dcef5dc86a07d678d5accf8dce97ac1738480ea783b71f7d740ea25a53763f485a08935b4dcc2d3feef8a0 SHA512 04eee304cc7e47d9a456d0a83645259c99f956019d11a8672a5eb582c26cf4d053fb5d1b613b48600b2e0aa42d06760b06a9b01dcc185ea5d3ba55af07faa4c9
MISC metadata.xml 335 BLAKE2B 06a96188c26e286e81590830a24987e0029dec21d88d4f9c6262fdddc4953f9dfedf0d7f66385e9a1e1465b2f32fc09cee72b52516a77e2ff4e24fa5fa380c6a SHA512 4762cebead407dcd66c274ff1abdd056fb1f14dc54603d476e22c462c9bcff8ab6b42baf9d00a6b937b4d36112112671aeddd1f493b7202fb02158d32817882a
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
+}