diff options
author | Pariksheet Nanda <pnanda@umich.edu> | 2023-06-04 19:28:41 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-06-04 19:28:41 -0400 |
commit | c08f9fd6fc9fda92e1b032f095f7699b64c585bf (patch) | |
tree | 763b34998f879b3991c019efddcd3b2d2b244652 | |
parent | c3fb998414881adc8af39874718e89c8a928d44f (diff) | |
download | spack-c08f9fd6fc9fda92e1b032f095f7699b64c585bf.tar.gz spack-c08f9fd6fc9fda92e1b032f095f7699b64c585bf.tar.bz2 spack-c08f9fd6fc9fda92e1b032f095f7699b64c585bf.tar.xz spack-c08f9fd6fc9fda92e1b032f095f7699b64c585bf.zip |
py-jaxlib: patch bazel server cpu limit (#37548) (#38090)
-rw-r--r-- | var/spack/repos/builtin/packages/py-jaxlib/package.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-jaxlib/package.py b/var/spack/repos/builtin/packages/py-jaxlib/package.py index 7d0c8b99da..d8def42008 100644 --- a/var/spack/repos/builtin/packages/py-jaxlib/package.py +++ b/var/spack/repos/builtin/packages/py-jaxlib/package.py @@ -51,6 +51,16 @@ class PyJaxlib(PythonPackage, CudaPackage): self.tmp_path = tempfile.mkdtemp(prefix="spack") self.buildtmp = tempfile.mkdtemp(prefix="spack") filter_file( + "build --spawn_strategy=standalone", + f""" +# Limit CPU workers to spack jobs instead of using all HOST_CPUS. +build --spawn_strategy=standalone +build --local_cpu_resources={make_jobs} +""".strip(), + ".bazelrc", + string=True, + ) + filter_file( 'f"--output_path={output_path}",', 'f"--output_path={output_path}",' f' "--sources_path={self.tmp_path}",' |