mirror of
https://github.com/pentoo/pentoo-overlay
synced 2025-12-28 11:13:10 +01:00
37 lines
1,022 B
Bash
37 lines
1,022 B
Bash
# Copyright 1999-2022 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=8
|
|
|
|
PYTHON_COMPAT=( python3_{10..11} )
|
|
|
|
inherit distutils-r1
|
|
|
|
#brach 22.04-dev
|
|
HASH_COMMIT="${PV}-dev"
|
|
|
|
DESCRIPTION="NVIDIA-maintained utilities to streamline mixed precision and distributed training in Pytorch"
|
|
HOMEPAGE="https://github.com/NVIDIA/apex"
|
|
SRC_URI="https://github.com/NVIDIA/apex/archive/${HASH_COMMIT}.tar.gz -> ${P}-gh.tar.gz"
|
|
|
|
LICENSE=""
|
|
SLOT="0"
|
|
KEYWORDS="amd64 ~arm64 x86"
|
|
IUSE=""
|
|
|
|
RDEPEND=">=dev-python/tqdm-4.28.1[${PYTHON_USEDEP}]
|
|
>=dev-python/numpy-1.15.3[${PYTHON_USEDEP}]
|
|
>=dev-python/pyyaml-5.1[${PYTHON_USEDEP}]"
|
|
DEPEND="${RDEPEND}"
|
|
REQUIRED_USE="${PYTHON_REQUIRED_USE}"
|
|
|
|
S="${WORKDIR}/apex-${HASH_COMMIT}"
|
|
|
|
#src_compile(){
|
|
# pip install -v --disable-pip-version-check --no-cache-dir ./
|
|
#pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
|
|
#}
|
|
|
|
python_compile() {
|
|
distutils-r1_python_compile --cpp_ext --cuda_ext
|
|
}
|