pystella/multigrid/__init__.py
def transfer_down(self, queue, i):
        """
        Transfers all arrays from a fine to the next-coarser level.

        :arg queue: A :class:`pyopencl.CommandQueue`.

        :arg i: The level from to transfer to.
        """

        for key, f1 in self.unknowns[i-1].items():
            f2 = self.unknowns[i][key]
            self.restrict(queue, f1=f1, f2=f2)
            self.decomp[i].share_halos(queue, f2)

        self.solver.residual(queue, **self.resid_args[i-1])

        for key, r1 in self.resid[i-1].items():
            r2 = self.resid[i][key]
            self.decomp[i-1].share_halos(queue, r1)
            self.restrict(queue, f1=r1, f2=r2)

        self.solver.lhs_correction(queue, **self.resid_args[i])
        for _, rho in self.rhos[i].items():
            self.decomp[i].share_halos(queue, rho)

        Transfers all arrays from a fine to the next-coarser level.

        :arg queue: A :class:`pyopencl.CommandQueue`.

        :arg i: The level from to transfer to.
        
Transfers all arrays from a fine to the next-coarser level.
Transfers all arrays from a fine to the next - coarser level .
def transfer_down(self, queue, i):
        for key, f1 in self.unknowns[i-1].items():
            f2 = self.unknowns[i][key]
            self.restrict(queue, f1=f1, f2=f2)
            self.decomp[i].share_halos(queue, f2)
        self.solver.residual(queue, **self.resid_args[i-1])
        for key, r1 in self.resid[i-1].items():
            r2 = self.resid[i][key]
            self.decomp[i-1].share_halos(queue, r1)
            self.restrict(queue, f1=r1, f2=r2)
        self.solver.lhs_correction(queue, **self.resid_args[i])
        for _, rho in self.rhos[i].items():
            self.decomp[i].share_halos(queue, rho)
def transfer_down ( self , queue , i ) : for key , f1 in self . unknowns [ i - 1 ] . items ( ) : f2 = self . unknowns [ i ] [ key ] self . restrict ( queue , f1 = f1 , f2 = f2 ) self . decomp [ i ] . share_halos ( queue , f2 ) self . solver . residual ( queue , ** self . resid_args [ i - 1 ] ) for key , r1 in self . resid [ i - 1 ] . items ( ) : r2 = self . resid [ i ] [ key ] self . decomp [ i - 1 ] . share_halos ( queue , r1 ) self . restrict ( queue , f1 = r1 , f2 = r2 ) self . solver . lhs_correction ( queue , ** self . resid_args [ i ] ) for _ , rho in self . rhos [ i ] . items ( ) : self . decomp [ i ] . share_halos ( queue , rho )
Transfers all arrays from a fine to the next-coarser level.
Transfers all arrays from a fine to the next - coarser level .
"""
        Transfers all arrays from a fine to the next-coarser level.

        :arg queue: A :class:`pyopencl.CommandQueue`.

        :arg i: The level from to transfer to.
        """
self
queue
None
i
The level from to transfer to.
The level from to transfer to .