def test_boundaries(): """ Tests if boundaries can be parallelized (or not). """ dimensions = 3 def reference_vector(vec): first_nonzero_position = next(i for i in range(len(vec)) if vec[i]) return -vec * Vector.unit_vector(first_nonzero_position, len(vec)) def get_stencil(boundary): # boundaries of 1-norm n depend on boundaries of 1-norm n-1 by looking in on the first non-zero element. read_vector = reference_vector(boundary) component = StencilComponent( 'mesh', SparseWeightArray( { read_vector: 1 } ) ) return Stencil(component, 'mesh', [ (-1, 0, 1) if bound == 1 else (0, 1, 1) if bound == -1 else (1, -1, 1) for bound in boundary ]) stencils = [ (boundary, get_stencil(boundary)) for boundary in Vector.moore_vectors(dimensions) ] group = StencilGroup([stencil for bound, stencil in stencils]) serial = create_dependency_graph(group, {"mesh": (32,)*dimensions}) parallel = create_parallel_graph(group, {"mesh": (32,)*dimensions}) # print(serial) # print(parallel) # print(serial == parallel) # exit() for (b1, s1), (b2, s2) in itertools.product(stencils, repeat=2): has_conflict = b2+reference_vector(b2) == b1 or b1 == b2 graph_conflict = serial[hash(s2)][hash(s1)] parallel_conflict = parallel[hash(s2)][hash(s1)] reported = stencil_conflict(s1, s2, shape_map={"mesh": (32,)*dimensions}) print(b1, b2, reported, has_conflict, graph_conflict, parallel_conflict, '*'*10*(parallel_conflict != has_conflict))