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

Use CudaPackage mixin for py-theano (#14577)

Browse source 
This PR adds the CudaPackage mixin class to py-theano. This replaces the
`gpu` variant with the `cuda` variant.
This commit is contained in:
Glenn Johnson 2020-01-21 09:27:41 -06:00 committed by Adam J. Stewart
parent d10505678f
commit 5db5040871
1 changed files with 5 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
var/spack/repos/builtin/packages/py-theano/package.py
View file

@ -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')