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