From 5db50408710a1a16a43099a4079f37e2af3cb548 Mon Sep 17 00:00:00 2001 From: Glenn Johnson Date: Tue, 21 Jan 2020 09:27:41 -0600 Subject: [PATCH] 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. --- .../repos/builtin/packages/py-theano/package.py | 13 +++++-------- 1 file 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')