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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (45) hide show
  1. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PKG-INFO +1 -1
  2. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/PKG-INFO +1 -1
  3. pyquda_utils-0.10.8.dev5/pyquda_utils/_version.py +1 -0
  4. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/core.py +9 -4
  5. pyquda_utils-0.10.8.dev3/pyquda_utils/_version.py +0 -1
  6. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/LICENSE +0 -0
  7. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/MANIFEST.in +0 -0
  8. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/SOURCES.txt +0 -0
  9. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/dependency_links.txt +0 -0
  10. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/entry_points.txt +0 -0
  11. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/requires.txt +0 -0
  12. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/PyQUDA_Utils.egg-info/top_level.txt +0 -0
  13. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/README.md +0 -0
  14. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyproject.toml +0 -0
  15. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/__init__.py +0 -0
  16. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/_field_utils.py +0 -0
  17. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/_mpi_file.py +0 -0
  18. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/chroma.py +0 -0
  19. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/io_general.py +0 -0
  20. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/kyu.py +0 -0
  21. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/lime.py +0 -0
  22. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/milc.py +0 -0
  23. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/nersc.py +0 -0
  24. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/npy.py +0 -0
  25. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/openqcd.py +0 -0
  26. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_io/xqcd.py +0 -0
  27. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_plugins/__main__.py +0 -0
  28. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_plugins/plugin_pyx.py +0 -0
  29. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_plugins/pygwu/__init__.py +0 -0
  30. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/__init__.py +0 -0
  31. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/alg_remez.py +0 -0
  32. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/convert.py +0 -0
  33. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/deprecated.py +0 -0
  34. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/eigensolve.py +0 -0
  35. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/gamma.py +0 -0
  36. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/gauge_nd_sun.py +0 -0
  37. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/gpt.py +0 -0
  38. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/hmc_param.py +0 -0
  39. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/io/__init__.py +0 -0
  40. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/milc_rhmc_param.py +0 -0
  41. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/phase.py +0 -0
  42. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/quasi_axial_gauge_fixing.py +0 -0
  43. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/pyquda_utils/source.py +0 -0
  44. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/setup.cfg +0 -0
  45. {pyquda_utils-0.10.8.dev3 → pyquda_utils-0.10.8.dev5}/setup.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: PyQUDA-Utils
3
- Version: 0.10.8.dev3
3
+ Version: 0.10.8.dev5
4
4
  Summary: Utility scripts based on PyQUDA
5
5
  Author-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
6
6
  Maintainer-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: PyQUDA-Utils
3
- Version: 0.10.8.dev3
3
+ Version: 0.10.8.dev5
4
4
  Summary: Utility scripts based on PyQUDA
5
5
  Author-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
6
6
  Maintainer-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
@@ -0,0 +1 @@
1
+ __version__ = "0.10.8.dev5"
@@ -127,10 +127,7 @@ def invertStaggeredPropagator(
127
127
 
128
128
 
129
129
  def gatherLattice2(
130
- data: numpy.ndarray,
131
- tzyx: List[int],
132
- reduce_op: Literal["sum", "mean", "prod", "max", "min"] = "sum",
133
- root: int = 0,
130
+ data: numpy.ndarray, tzyx: List[int], reduce_op: Literal["sum", "mean", "prod", "max", "min"] = "sum", root: int = 0
134
131
  ):
135
132
  sendobj = numpy.ascontiguousarray(data)
136
133
  recvobj = getMPIComm().gather(sendobj, root)
@@ -203,6 +200,14 @@ def scatterLattice(data: numpy.ndarray, tzyx: List[int], root: int = 0):
203
200
  return recvobj
204
201
 
205
202
 
203
+ def gatherScatterLattice(
204
+ data: numpy.ndarray, tzyx: List[int], reduce_op: Literal["sum", "mean", "prod", "max", "min"] = "sum", root: int = 0
205
+ ):
206
+ data = gatherLattice2(data, tzyx, reduce_op, root)
207
+ data = scatterLattice(data, tzyx, root)
208
+ return data
209
+
210
+
206
211
  def gatherLattice(data: numpy.ndarray, axes: List[int], reduce_op: Literal["sum", "mean"] = "sum", root: int = 0):
207
212
  """
208
213
  MPI gather or reduce data from all MPI subgrid onto the root process.
@@ -1 +0,0 @@
1
- __version__ = "0.10.8.dev3"