Skip to content

Commit fe3dc30

Browse files
authored
Merge pull request lisa-lab#180 from nouiz/master2
Do the speed test on the new gpu back-end.
2 parents a74f528 + fd5cb65 commit fe3dc30

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

code/test.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -190,9 +190,9 @@ def do_tests():
190190
float64_times / float32_times), file=sys.stderr)
191191

192192
#test in float32 in FAST_RUN mode on the gpu
193-
import theano.sandbox.cuda
193+
import theano.gpuarray
194194
if do_gpu:
195-
theano.sandbox.cuda.use('gpu')
195+
theano.gpuarray.use('cuda')
196196
gpu_times = do_tests()
197197
times_dic['gpu'] = gpu_times
198198
test_total += numpy.size(gpu_times)

0 commit comments

Comments
 (0)