Adapt sphinx configuration for new gpu backend

Lasagne · Jul 12, 2017 · 1aa7756 · 1aa7756
1 parent 164c9d3
commit 1aa7756
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -334,6 +334,12 @@ def setup(app):
 theano.sandbox.cuda.cuda_enabled = True
 theano.sandbox.cuda.dnn.dnn_available = lambda: True
 
+theano.gpuarray = Mock()
+sys.modules['theano.gpuarray'] = theano.gpuarray
+sys.modules['theano.gpuarray.dnn'] = theano.gpuarray.dnn
+theano.gpuarray.pygpu_activated = True
+theano.gpuarray.dnn.dnn_present = lambda: True
+
 sys.modules['pylearn2'] = Mock()
 sys.modules['pylearn2.sandbox'] = Mock()
 sys.modules['pylearn2.sandbox.cuda_convnet'] = Mock()