Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.5 kB · 72 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273{ lib, buildPythonPackage, fetchFromGitHub, pythonOlder, setuptools# build inputs, torch, numpy, ninja# check inputs, pytestCheckHook, parameterized, pytest-cov, pytest-timeout, remote-pdb}:let pname = "fairscale"; version = "0.4.13";inbuildPythonPackage { inherit pname version; format = "pyproject";
disabled = pythonOlder "3.10";
src = fetchFromGitHub { owner = "facebookresearch"; repo = "fairscale"; rev = "refs/tags/v${version}"; hash = "sha256-L2Rl/qL6l0OLAofygzJBGQdp/2ZrgDFarwZRjyAR3dw="; };
# setup.py depends on ninja python dependency, but we have the binary in nixpkgs postPatch = '' substituteInPlace setup.py \ --replace 'setup_requires=["ninja"]' 'setup_requires=[]' '';
nativeBuildInputs = [ ninja setuptools ];
propagatedBuildInputs = [ torch numpy ];
nativeCheckInputs = [ pytestCheckHook parameterized pytest-cov pytest-timeout remote-pdb ];
# Some tests try to build distributed models, which doesn't work in the sandbox. doCheck = false;
pythonImportsCheck = [ "fairscale" ];
meta = with lib; { description = "PyTorch extensions for high performance and large scale training"; homepage = "https://github.com/facebookresearch/fairscale"; changelog = "https://github.com/facebookresearch/fairscale/releases/tag/v${version}"; license = with licenses; [ mit asl20 bsd3 ]; maintainers = with maintainers; [ happysalada ]; };}