mirror of
https://github.com/pentoo/pentoo-overlay
synced 2025-12-06 16:33:09 +01:00
55 lines
1.4 KiB
Bash
55 lines
1.4 KiB
Bash
# Copyright 1999-2023 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=8
|
|
|
|
DISTUTILS_USE_PEP517=setuptools
|
|
PYTHON_COMPAT=( python3_{10..12} )
|
|
|
|
#FIXME: sci-libs/pytorch is single package:
|
|
#DISTUTILS_SINGLE_IMPL=1
|
|
#inherit python-single-r1
|
|
|
|
inherit distutils-r1
|
|
|
|
DESCRIPTION="Tools for easy mixed precision and distributed training in Pytorch"
|
|
HOMEPAGE="https://github.com/NVIDIA/apex"
|
|
SRC_URI="https://github.com/NVIDIA/apex/archive/refs/tags/${PV}.tar.gz -> ${P}-gh.tar.gz"
|
|
|
|
LICENSE="BSD"
|
|
SLOT="0"
|
|
KEYWORDS="~amd64"
|
|
|
|
IUSE="cuda"
|
|
|
|
#FIXME: add --pyprof
|
|
#https://github.com/NVIDIA/PyProf
|
|
|
|
RDEPEND=">=dev-python/cxxfilt-0.2.0[${PYTHON_USEDEP}]
|
|
>=dev-python/tqdm-4.28.1[${PYTHON_USEDEP}]
|
|
>=dev-python/numpy-1.15.3[${PYTHON_USEDEP}]
|
|
>=sci-libs/pytorch-1.12.0
|
|
>=dev-python/pyyaml-5.1[${PYTHON_USEDEP}]
|
|
>=dev-python/packaging-14.0[${PYTHON_USEDEP}]"
|
|
DEPEND="${RDEPEND}"
|
|
REQUIRED_USE="${PYTHON_REQUIRED_USE}"
|
|
|
|
RESTRICT="test"
|
|
|
|
S="${WORKDIR}/apex-${PV}"
|
|
|
|
#If you wish to cross-compile for a single specific architecture,
|
|
#export TORCH_CUDA_ARCH_LIST="compute capability" before running setup.py.
|
|
python_configure_all() {
|
|
if use cuda; then
|
|
# export MAX_JOBS=1
|
|
#export TORCH_CUDA_ARCH_LIST="compute capability"
|
|
export TORCH_CUDA_ARCH_LIST="7.5"
|
|
|
|
DISTUTILS_ARGS=( --cpp_ext --cuda_ext )
|
|
fi
|
|
}
|
|
|
|
python_compile() {
|
|
distutils-r1_python_compile -j1
|
|
}
|