943c8091c2
* simple build of py-openai * added variants to py-openai * py-pandas-stubs is a dependency for py-openai * fixed format and flake8 errors for py-openai * black format error for py-pandas-stubs * [@spackbot] updating style on behalf of sidpbury * made style and format changes to py-openai * made style and format changes to py-pandas-stubs * py-types-pytz is a dependency for py-openai * [@spackbot] updating style on behalf of sidpbury * updated py-openpyxl for ver 3.0.7 and 3.1.2 * Update var/spack/repos/builtin/packages/py-pandas-stubs/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> * ajs requested changes for py-openai * updated py-openpyxl for supported python * [@spackbot] updating style on behalf of sidpbury * updated py-openpyxl * removed requirement.txt dependencies in py-openpyxl * removed python depends on from openpyxl * updated package to support newer versions * updated version of py-pygit2 * py-fairscale is a new package in support of torch * py-pgzip is a dependency for py-fairscale * switch fairscale pypi, added extra variant for convenience * removed python dependency * changed multiple requirement versions * changes for upstream py-fairscale * changes for upsteam py-pygit2 * Update package.py * Update package.py * sorted out some of the dependency versions * removed version 1.12.2 because dependency could not be met * updated py-cached-property dependency * suggested changes from adamjstewart --------- Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
38 lines
1.6 KiB
Python
38 lines
1.6 KiB
Python
# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
from spack.package import *
|
|
|
|
|
|
class PyFairscale(PythonPackage):
|
|
"""FairScale is a PyTorch extension library for high performance and large
|
|
scale training. This library extends basic PyTorch capabilities while adding
|
|
new SOTA scaling techniques."""
|
|
|
|
homepage = "https://github.com/facebookresearch/fairscale"
|
|
pypi = "fairscale/fairscale-0.4.13.tar.gz"
|
|
|
|
version("0.4.13", sha256="1b797825c427f5dba92253fd0d8daa574e8bd651a2423497775fab1b30cfb768")
|
|
version("0.4.4", sha256="7719898743dc58c04a2294c896ee6308c92ccb3af9e10632b2a62f77cb689357")
|
|
|
|
variant("extra", default=False, description="support for cuda.list.gpu, scaler, weight")
|
|
|
|
# from setup.py
|
|
depends_on("python@3.8:", type=("build", "run"))
|
|
depends_on("ninja", type="build")
|
|
# from pyproject.toml
|
|
depends_on("py-wheel@0.30.0:", type="build")
|
|
depends_on("py-setuptools@40.6.2:", type="build")
|
|
# from requirements.txt
|
|
depends_on("py-torch@1.8.0:", type=("build", "run"))
|
|
depends_on("py-numpy@1.22.0:", type=("build", "run"))
|
|
# added extra to support cuda.list.gpu, scaler, and weight (not in pip install)
|
|
# see requirements-dev.txt in github. This does not include the tools for testing
|
|
with when("+extra"):
|
|
depends_on("py-pynvml@8.0.4", type=("build", "run"))
|
|
depends_on("py-scikit-learn@1.1.3", type=("build", "run"))
|
|
depends_on("py-pygit2@1.11.1", type=("build", "run"))
|
|
depends_on("py-pgzip@0.3.1", type=("build", "run"))
|