def run_test(): size = 130 weight_array = WeightArray( np.ones((3, 3), dtype=np.float) / 9 ) component = StencilComponent( "input", weight_array ) red = DomainUnion([ RectangularDomain(( (1, -1, 2), (1, -1, 2) )), RectangularDomain(( (2, 0, 2), (2, 0, 2) )) ]) stencil = Stencil( component, "output", red ) ccompiler = OpenMPCompiler() kern = ccompiler.compile(stencil) arr = np.arange(size**2, dtype=np.float).reshape((size, size)) out = np.zeros_like(arr) kern(out, arr) print(out)
def run_test(): weight_array = WeightArray(np.ones((3, 3, 3), dtype=np.float) / 27) component = StencilComponent("input", weight_array) stencil = Stencil(component, "output", [(1, -1, 1)] * 3) ccompiler = OpenMPCompiler() kern = ccompiler.compile(stencil) arr = np.arange(66**3, dtype=np.float).reshape((66, 66, 66)) out = np.zeros_like(arr) kern(out, arr)
def run_test(): size = 130 weight_array = WeightArray(np.ones((3, 3), dtype=np.float) / 9) component = StencilComponent("input", weight_array) red = DomainUnion([ RectangularDomain(((1, -1, 2), (1, -1, 2))), RectangularDomain(((2, 0, 2), (2, 0, 2))) ]) stencil = Stencil(component, "output", red) ccompiler = OpenMPCompiler() kern = ccompiler.compile(stencil) arr = np.arange(size**2, dtype=np.float).reshape((size, size)) out = np.zeros_like(arr) kern(out, arr) print(out)
def run_test(): weight_array = WeightArray( np.ones((3, 3, 3), dtype=np.float) / 27 ) component = StencilComponent( "input", weight_array ) stencil = Stencil( component, "output", [(1, -1, 1)]*3 ) ccompiler = OpenMPCompiler() kern = ccompiler.compile(stencil) arr = np.arange(66**3, dtype=np.float).reshape((66, 66, 66)) out = np.zeros_like(arr) kern(out, arr)