diff options
| author | Auriane R <48684432+aurianer@users.noreply.github.com> | 2024-03-23 16:04:10 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-23 16:04:10 +0100 |
| commit | 99bb288db7a62d35654af7fa275258338a8d14d8 (patch) | |
| tree | 80cf301ef4e331a4ab22f065d5c07e0b569a4fec | |
| parent | 99744a766bb266654a96a796f5841c401aeb8e25 (diff) | |
| download | spack-99bb288db7a62d35654af7fa275258338a8d14d8.tar.gz spack-99bb288db7a62d35654af7fa275258338a8d14d8.tar.bz2 spack-99bb288db7a62d35654af7fa275258338a8d14d8.tar.xz spack-99bb288db7a62d35654af7fa275258338a8d14d8.zip | |
py-torch-nvidia-apex: @3.11: Add config_settings(PEP517), add missing py-packaging (#43306)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py b/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py index 45f9b279cb..00a357603f 100644 --- a/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py +++ b/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py @@ -21,6 +21,7 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): depends_on("python@3:", type=("build", "run")) depends_on("py-setuptools", type="build") + depends_on("py-packaging", type="build") depends_on("py-torch@0.4:", type=("build", "run")) depends_on("cuda@9:", when="+cuda") depends_on("py-pybind11", type=("build", "link", "run")) @@ -43,6 +44,7 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): else: env.unset("CUDA_HOME") + @when("^python@:3.10") def global_options(self, spec, prefix): args = [] if spec.satisfies("^py-torch@1.0:"): @@ -50,3 +52,11 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): if "+cuda" in spec: args.append("--cuda_ext") return args + + @when("^python@3.11:") + def config_settings(self, spec, prefix): + return { + "builddir": "build", + "compile-args": f"-j{make_jobs}", + "--global-option": "--cpp_ext --cuda_ext", + } |