def local_to_global_end(self, dat, insert_mode): assert insert_mode in {op2.INC, op2.MIN, op2.MAX}, "%s LtoG not supported" % insert_mode if self.comm.size == 1: return mtype, builtin = _get_mtype(dat) op = {(False, op2.INC): MPI.SUM, (True, op2.INC): MPI.SUM, (False, op2.MIN): _contig_min_op, (True, op2.MIN): MPI.MIN, (False, op2.MAX): _contig_max_op, (True, op2.MAX): MPI.MAX}[(builtin, insert_mode)] dmplex.halo_end(self.sf, dat, mtype, True, op=op)
def local_to_global_end(self, dat, insert_mode): assert insert_mode in {op2.INC, op2.MIN, op2.MAX}, "%s LtoG not supported" % insert_mode if self.comm.size == 1: return mtype = _get_mtype(dat) op = { op2.INC: MPI.SUM, op2.MIN: MPI.MIN, op2.MAX: MPI.MAX }[insert_mode] dmplex.halo_end(self.sf, dat, mtype, True, op=op)
def end(self, dat, reverse=False): """End a halo exchange. :arg dat: The :class:`pyop2.Dat` to end a halo exchange on. :arg reverse: (optional) perform a reverse halo exchange. See also :meth:`.begin`.""" if self.comm.size == 1: return mtype = _get_mtype(dat) maybe_setflags(dat._data, write=True) dmplex.halo_end(self.sf, dat, mtype, reverse) maybe_setflags(dat._data, write=False)
def end(self, dat, reverse=False): """End a halo exchange. :arg dat: The :class:`pyop2.Dat` to end a halo exchange on. :arg reverse: (optional) perform a reverse halo exchange. See also :meth:`.begin`.""" if self.comm.size == 1: return mtype = _get_mtype(dat) maybe_setflags(dat._data, write=True) dmplex.halo_end(self.sf, dat, mtype, reverse) maybe_setflags(dat._data, write=False)
def local_to_global_end(self, dat, insert_mode): assert insert_mode in {op2.INC, op2.MIN, op2.MAX}, "%s LtoG not supported" % insert_mode if self.comm.size == 1: return mtype, builtin = _get_mtype(dat) op = { (False, op2.INC): MPI.SUM, (True, op2.INC): MPI.SUM, (False, op2.MIN): _contig_min_op, (True, op2.MIN): MPI.MIN, (False, op2.MAX): _contig_max_op, (True, op2.MAX): MPI.MAX }[(builtin, insert_mode)] dmplex.halo_end(self.sf, dat, mtype, True, op=op)
def global_to_local_end(self, dat, insert_mode): assert insert_mode is op2.WRITE, "Only WRITE GtoL supported" if self.comm.size == 1: return mtype, _ = _get_mtype(dat) dmplex.halo_end(self.sf, dat, mtype, False)
def global_to_local_end(self, dat, insert_mode): assert insert_mode is op2.WRITE, "Only WRITE GtoL supported" if self.comm.size == 1: return mtype, _ = _get_mtype(dat) dmplex.halo_end(self.sf, dat, mtype, False)