diff options
author | Glenn Johnson <glenn-johnson@uiowa.edu> | 2020-01-21 09:27:41 -0600 |
---|---|---|
committer | Adam J. Stewart <ajstewart426@gmail.com> | 2020-01-21 09:27:41 -0600 |
commit | 5db50408710a1a16a43099a4079f37e2af3cb548 (patch) | |
tree | f8ff48cd699154e6d1450db6d27fe360f2dc5171 /var | |
parent | d10505678fd5624e5e88f72ac7852109f149b264 (diff) | |
download | spack-5db50408710a1a16a43099a4079f37e2af3cb548.tar.gz spack-5db50408710a1a16a43099a4079f37e2af3cb548.tar.bz2 spack-5db50408710a1a16a43099a4079f37e2af3cb548.tar.xz spack-5db50408710a1a16a43099a4079f37e2af3cb548.zip |
Use CudaPackage mixin for py-theano (#14577)
This PR adds the CudaPackage mixin class to py-theano. This replaces the
`gpu` variant with the `cuda` variant.
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/py-theano/package.py | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/var/spack/repos/builtin/packages/py-theano/package.py b/var/spack/repos/builtin/packages/py-theano/package.py index cbe496c3ed..c7c95fd0e3 100644 --- a/var/spack/repos/builtin/packages/py-theano/package.py +++ b/var/spack/repos/builtin/packages/py-theano/package.py @@ -6,7 +6,7 @@ from spack import * -class PyTheano(PythonPackage): +class PyTheano(PythonPackage, CudaPackage): """Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.""" @@ -20,9 +20,6 @@ class PyTheano(PythonPackage): version('1.0.1', sha256='88d8aba1fe2b6b75eacf455d01bc7e31e838c5a0fb8c13dde2d9472495ff4662') version('0.8.2', sha256='7463c8f7ed1a787bf881f36d38a38607150186697e7ce7e78bfb94b7c6af8930') - variant('gpu', default=False, - description='Builds with support for GPUs via CUDA and cuDNN') - depends_on('python@2.6:2.8,3.3:') depends_on('py-setuptools', type=('build', 'run')) @@ -32,10 +29,10 @@ class PyTheano(PythonPackage): depends_on('blas') - depends_on('cuda', when='+gpu') - depends_on('cudnn', when='+gpu') - depends_on('py-pygpu', when='+gpu', type=('build', 'run')) - depends_on('libgpuarray', when='+gpu') + depends_on('cuda', when='+cuda') + depends_on('cudnn', when='+cuda') + depends_on('py-pygpu', when='+cuda', type=('build', 'run')) + depends_on('libgpuarray', when='+cuda') depends_on('py-nose@1.3.0:', type='test') depends_on('py-parameterized', type='test') |