提交 30df2588 authored 作者: Frederic Bastien's avatar Frederic Bastien

faster test and test not square image for ConvOp.

上级 6eb6bfd0
...@@ -704,12 +704,18 @@ class CLinker(link.Linker): ...@@ -704,12 +704,18 @@ class CLinker(link.Linker):
instantiate.customize.add_support_code(self.struct_code) instantiate.customize.add_support_code(self.struct_code)
instantiate.customize.add_support_code(static) instantiate.customize.add_support_code(static)
for extra_arg in ( for extra_arg in (
"-O2", "-O3",
# "-fno-signaling-nans",
#"-fno-finite-math-only",
#"-fmath-errno", "-fno-unsafe-math-optimizations", "-fno-finite-math-only", "-frounding-math", "-fsignaling-nans","-fno-cx-limited-range","-fno-fast-math",
"-ffast-math", "-ffast-math",
#"-fno-finite-math-only",
# "-fno-signaling-nans",
#"-fmath-errno", "-fno-unsafe-math-optimizations", "-fno-finite-math-only", "-frounding-math", "-fsignaling-nans","-fno-cx-limited-range","-fno-fast-math",
#"-fprefetch-loop-arrays", #"-fprefetch-loop-arrays",
#"-ftree-vect-loop-version", #"-ftree-vect-loop-version",
#"-ftree-loop-optimize", #"-ftree-loop-optimize",
#"-ftree-vectorize"): #"-ftree-vectorize",
"-w" #-w means supress all warnings "-w" #-w means supress all warnings
): ):
instantiate.customize.add_extra_compile_arg(extra_arg) instantiate.customize.add_extra_compile_arg(extra_arg)
......
...@@ -304,7 +304,7 @@ class TestConvOp(unittest.TestCase): ...@@ -304,7 +304,7 @@ class TestConvOp(unittest.TestCase):
def test_multilayer_conv(self): def test_multilayer_conv(self):
# fixed parameters # fixed parameters
bsizes = [6,6] # batch size bsizes = [6,6] # batch size
imshp_starts = [(1,28,28),(1,4,4)] imshp_starts = [(1,13,14),(1,4,3)]
kshpss = ([[5,6],[7,4]],[[2,2],[2,2]]) kshpss = ([[5,6],[7,4]],[[2,2],[2,2]])
nkernss = [[20,40],[2,2]] # per output pixel nkernss = [[20,40],[2,2]] # per output pixel
ssizess = [[(1,1),(2,2)],[(1,1),(2,2)]] ssizess = [[(1,1),(2,2)],[(1,1),(2,2)]]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论