hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

py-jaxlib: patch bazel server cpu limit (#37548) (#38090)

Browse source
This commit is contained in:
Pariksheet Nanda 2023-06-04 19:28:41 -04:00 committed by GitHub
parent c3fb998414
commit c08f9fd6fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
var/spack/repos/builtin/packages/py-jaxlib/package.py
View file

@ -50,6 +50,16 @@ class PyJaxlib(PythonPackage, CudaPackage):
def patch(self):
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}",'