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

it is on the gpu that the test should be done in 3d, not cpu.

上级 ed626d81
...@@ -855,8 +855,8 @@ class test_fusion(unittest.TestCase): ...@@ -855,8 +855,8 @@ class test_fusion(unittest.TestCase):
mode._optimizer=mode._optimizer.including('local_elemwise_fusion','canonicalize') mode._optimizer=mode._optimizer.including('local_elemwise_fusion','canonicalize')
self.do(mode, shared, shp) self.do(mode, shared, shp)
def test_elemwise_fusion_3d(self): def test_elemwise_fusion_4d(self):
shp=(3,3,3) shp=(3,3,3,3)
mode=copy.copy(compile.mode.get_default_mode()) mode=copy.copy(compile.mode.get_default_mode())
#we need the optimisation enabled and the canonicalize. #we need the optimisation enabled and the canonicalize.
#the canonicalize is needed to merge multiplication/addition by constant. #the canonicalize is needed to merge multiplication/addition by constant.
...@@ -876,8 +876,8 @@ class test_fusion(unittest.TestCase): ...@@ -876,8 +876,8 @@ class test_fusion(unittest.TestCase):
self.do(mode, cuda.float32_shared_constructor, shp, gpu=True) self.do(mode, cuda.float32_shared_constructor, shp, gpu=True)
def test_gpu_fusion_4d(self): def test_gpu_fusion_3d(self):
shp=(5,5,5,5) shp=(5,5,5)
#we need the optimisation enabled, debug do this. #we need the optimisation enabled, debug do this.
if theano.config.mode == "FAST_COMPILE": if theano.config.mode == "FAST_COMPILE":
mode = theano.compile.mode.get_mode("FAST_RUN").including('local_elemwise_fusion','canonicalize','gpu') mode = theano.compile.mode.get_mode("FAST_RUN").including('local_elemwise_fusion','canonicalize','gpu')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论