提交 6949fa64 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Adapted tests to use new version of GpuImages2Neibs op

上级 1bafd598
......@@ -6,7 +6,7 @@ if cuda_ndarray.cuda_available == False:
raise SkipTest('Optional package cuda disabled')
import theano.sandbox.test_neighbours
from theano.sandbox.cuda.neighbours import GpuImages2Neibs
from theano.sandbox.gpuarray.neighbours import GpuImages2Neibs
if theano.config.mode == 'FAST_COMPILE':
mode_with_gpu = theano.compile.mode.get_mode('FAST_RUN').including('gpu')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论