diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py | 6 |
1 files changed, 2 insertions, 4 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 af6666b6c8..4056c8fb25 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 @@ -11,8 +11,6 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): homepage = "https://github.com/nvidia/apex/" git = "https://github.com/nvidia/apex/" - phases = ['install'] - version('master', branch='master') version('2020-10-19', commit='8a1ed9e8d35dfad26fb973996319965e4224dcdd') @@ -34,8 +32,8 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): self.spec.variants['cuda_arch'].value) env.set('TORCH_CUDA_ARCH_LIST', torch_cuda_arch) - def install_args(self, spec, prefix): - args = super(PyTorchNvidiaApex, self).install_args(spec, prefix) + def install_options(self, spec, prefix): + args = [] if spec.satisfies('^py-torch@1.0:'): args.append('--cpp_ext') if '+cuda' in spec: |