PyQUDA-Utils 0.10.8.dev1__tar.gz → 0.10.8.dev5__tar.gz

{pyquda_utils-0.10.8.dev1 → pyquda_utils-0.10.8.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyQUDA-Utils
-Version: 0.10.8.dev1
+Version: 0.10.8.dev5
 Summary: Utility scripts based on PyQUDA
 Author-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
 Maintainer-email: SaltyChiang <SaltyChiang@users.noreply.github.com>

{pyquda_utils-0.10.8.dev1 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyQUDA-Utils
-Version: 0.10.8.dev1
+Version: 0.10.8.dev5
 Summary: Utility scripts based on PyQUDA
 Author-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
 Maintainer-email: SaltyChiang <SaltyChiang@users.noreply.github.com>

pyquda_utils-0.10.8.dev5/pyquda_utils/_version.py

@@ -0,0 +1 @@
+__version__ = "0.10.8.dev5"

{pyquda_utils-0.10.8.dev1 → pyquda_utils-0.10.8.dev5}/pyquda_utils/core.py

@@ -47,12 +47,11 @@ from pyquda.field import (  # noqa: F401
     LatticeStaggeredPropagator,
     lexico,
     evenodd,
-    evenodd as cb2,
 )
 from pyquda.dirac.abstract import Multigrid, FermionDirac, StaggeredFermionDirac
 
 from . import source
-from .deprecated import smear, smear4, invert12, getDslash, getStaggeredDslash
+from .deprecated import smear, smear4, invert12, getDslash, getStaggeredDslash, cb2
 
 LaplaceLatticeInfo = LatticeInfo
 
@@ -128,10 +127,7 @@ def invertStaggeredPropagator(
 
 
 def gatherLattice2(
-    data: numpy.ndarray,
-    tzyx: List[int],
-    reduce_op: Literal["sum", "mean", "prod", "max", "min"] = "sum",
-    root: int = 0,
+    data: numpy.ndarray, tzyx: List[int], reduce_op: Literal["sum", "mean", "prod", "max", "min"] = "sum", root: int = 0
 ):
     sendobj = numpy.ascontiguousarray(data)
     recvobj = getMPIComm().gather(sendobj, root)
@@ -204,6 +200,14 @@ def scatterLattice(data: numpy.ndarray, tzyx: List[int], root: int = 0):
     return recvobj
 
 
+def gatherScatterLattice(
+    data: numpy.ndarray, tzyx: List[int], reduce_op: Literal["sum", "mean", "prod", "max", "min"] = "sum", root: int = 0
+):
+    data = gatherLattice2(data, tzyx, reduce_op, root)
+    data = scatterLattice(data, tzyx, root)
+    return data
+
+
 def gatherLattice(data: numpy.ndarray, axes: List[int], reduce_op: Literal["sum", "mean"] = "sum", root: int = 0):
     """
     MPI gather or reduce data from all MPI subgrid onto the root process.

{pyquda_utils-0.10.8.dev1 → pyquda_utils-0.10.8.dev5}/pyquda_utils/deprecated.py

@@ -1,10 +1,17 @@
-from typing import List
+from typing import Any, List
+
+import numpy
 
 from pyquda import getLogger, getGridSize, pyquda as quda, enum_quda
-from pyquda.field import LatticeFermion, LatticeGauge, LatticeInfo, LatticePropagator, Nc, Ns
+from pyquda.field import LatticeFermion, LatticeGauge, LatticeInfo, LatticePropagator, Nc, Ns, evenodd
 from pyquda.dirac.abstract import FermionDirac
 
 
+def cb2(data: numpy.ndarray, axes: List[int], dtype=None):
+    getLogger().warning("Use evenodd instead", DeprecationWarning)
+    return evenodd(data, axes, dtype)
+
+
 def smear(latt_size: List[int], gauge: LatticeGauge, nstep: int, rho: float):
     getLogger().warning("Use GaugeField.stoutSmear instead", DeprecationWarning)
     from .core import getDslash

{pyquda_utils-0.10.8.dev1 → pyquda_utils-0.10.8.dev5}/pyquda_utils/quasi_axial_gauge_fixing.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from .core import getRankFromCoord, evenodd, LatticeLink, LatticeGauge, LatticeFermion, X, Y, Z, T
+from .core import getRankFromCoord, evenodd, LatticeGauge, LatticeFermion, X, Y, Z, T
 
 
 def quasiAxialGaugeFixing(gauge: LatticeGauge, dir: int):
@@ -57,14 +57,14 @@ def quasiAxialGaugeFixing(gauge: LatticeGauge, dir: int):
     rotate[0] = contract("xab,xb->xab", v, cupy.exp(1j * (gi * Li) / GLi * w))
     for i in range(1, Li):
         rotate[i] = contract("xba,xbc,xc->xac", gauge_prod[i - 1].conj(), v, cupy.exp(1j * (i + gi * Li) / GLi * w))
-    rotate = LatticeLink(gauge.latt_info, evenodd(rotate.reshape(*axes_shape).transpose(*axes).get(), [0, 1, 2, 3]))
+    rotate = LatticeGauge(gauge.latt_info, 1, evenodd(rotate.reshape(*axes_shape).transpose(*axes).get(), [0, 1, 2, 3]))
     rotate.toDevice()
     rotate_ = LatticeFermion(gauge.latt_info)
-    rotate.pack(rotate_)
+    rotate.pack(0, rotate_)
     gauge.data = contract("wtzyxba,dwtzyxbc->dwtzyxac", rotate.data.conj(), gauge.data)
     gauge.gauge_dirac.loadGauge(gauge)
-    gauge[X].unpack(gauge.gauge_dirac.covDev(rotate_, X))
-    gauge[Y].unpack(gauge.gauge_dirac.covDev(rotate_, Y))
-    gauge[Z].unpack(gauge.gauge_dirac.covDev(rotate_, Z))
-    gauge[T].unpack(gauge.gauge_dirac.covDev(rotate_, T))
+    gauge.unpack(X, gauge.gauge_dirac.covDev(rotate_, X))
+    gauge.unpack(Y, gauge.gauge_dirac.covDev(rotate_, Y))
+    gauge.unpack(Z, gauge.gauge_dirac.covDev(rotate_, Z))
+    gauge.unpack(T, gauge.gauge_dirac.covDev(rotate_, T))
     return rotate

pyquda_utils-0.10.8.dev1/pyquda_utils/_version.py

@@ -1 +0,0 @@
-__version__ = "0.10.8.dev1"