diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/py-theano/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/py-theano/package.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-theano/package.py b/var/spack/repos/builtin/packages/py-theano/package.py index 2806c00be3..3d64726511 100644 --- a/var/spack/repos/builtin/packages/py-theano/package.py +++ b/var/spack/repos/builtin/packages/py-theano/package.py @@ -34,12 +34,20 @@ class PyTheano(PythonPackage): version('0.8.2', 'f2d0dfe7df141115201077cd933b2c52') + 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') depends_on('py-scipy@0.11:', type=('build', 'run')) depends_on('py-numpy@1.7.1:', type=('build', 'run')) depends_on('py-six@1.9.0:', type=('build', 'run')) + + depends_on('blas') + + depends_on('cuda', when='+gpu') + depends_on('libgpuarray', when='+gpu') # test requirements # depends_on('py-nose@1.3.0:', type=('build', 'run')) # depends_on('py-nose-parameterized@0.5.0:', type=('build', 'run')) |