提交 b0199324 authored 作者: Vikram's avatar Vikram

Op replacment instead of calling Op directly

上级 e8e1dcef
...@@ -268,12 +268,7 @@ class TestGroupGpuCorr2d(Grouped_conv_noOptim): ...@@ -268,12 +268,7 @@ class TestGroupGpuCorr2d(Grouped_conv_noOptim):
class TestUnsharedGpuCorr2d(TestUnsharedConv): class TestUnsharedGpuCorr2d(TestUnsharedConv):
mode = theano.compile.get_mode("FAST_RUN") mode = mode_with_gpu
conv2d = GpuCorrMM
conv2d_gradw = GpuCorrMM_gradWeights
conv2d_gradi = GpuCorrMM_gradInputs
conv2d_op = GpuCorrMM conv2d_op = GpuCorrMM
conv2d_gradw_op = GpuCorrMM_gradWeights conv2d_gradw_op = GpuCorrMM_gradWeights
conv2d_gradi_op = GpuCorrMM_gradInputs conv2d_gradi_op = GpuCorrMM_gradInputs
flip_filter = True
is_dnn = False
...@@ -454,17 +454,12 @@ class TestGroupCorr2d(Grouped_conv_noOptim): ...@@ -454,17 +454,12 @@ class TestGroupCorr2d(Grouped_conv_noOptim):
class TestUnsharedCorr2d(TestUnsharedConv): class TestUnsharedCorr2d(TestUnsharedConv):
if theano.config.mode == "FAST_COMPILE": if theano.config.mode == "FAST_COMPILE":
mode = theano.compile.get_mode("FAST_RUN") mode = theano.compile.get_mode("FAST_RUN").excluding('gpuarray')
else: else:
mode = None mode = None
conv2d = corr.CorrMM
conv2d_gradw = corr.CorrMM_gradWeights
conv2d_gradi = corr.CorrMM_gradInputs
conv2d_op = corr.CorrMM conv2d_op = corr.CorrMM
conv2d_gradw_op = corr.CorrMM_gradWeights conv2d_gradw_op = corr.CorrMM_gradWeights
conv2d_gradi_op = corr.CorrMM_gradInputs conv2d_gradi_op = corr.CorrMM_gradInputs
flip_filter = True
is_dnn = False
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论