提交 ed52e669 authored 作者: Frederic Bastien's avatar Frederic Bastien

Force some reference impl in test in FAST_RUN. Fix test error in mode=FAST_COMPILE.

上级 15d5e390
......@@ -147,12 +147,12 @@ class TestCorr3DMM(unittest.TestCase):
conv_gemm = GpuCorr3dMM_gradWeights(subsample=subsample)(
img, topgrad)
else:
conv_ref = GpuCorr3dMM_gradWeights(subsample=subsample)(
conv_ref = Corr3dMM_gradWeights(subsample=subsample)(
img, topgrad, shape=filters_shape[1:4])
conv_gemm = GpuCorr3dMM_gradWeights(subsample=subsample)(
img, topgrad, shape=filters_shape[1:4])
f_ref = theano.function([], conv_ref)
f_ref = theano.function([], conv_ref, mode='FAST_RUN')
f = theano.function([], conv_gemm, mode=mode_with_gpu)
res_ref = f_ref()
......@@ -206,7 +206,7 @@ class TestCorr3DMM(unittest.TestCase):
kern=weight, topgrad=top,
shape=bottom_shape)
f_ref = theano.function([], conv_ref)
f_ref = theano.function([], conv_ref, mode='FAST_RUN')
f = theano.function([], conv_gemm, mode=mode_with_gpu)
res_ref = f_ref()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论