PyQUDA-Utils 0.10.7.dev3__tar.gz → 0.10.8.dev0__tar.gz

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyQUDA-Utils
-Version: 0.10.7.dev3
+Version: 0.10.8.dev0
 Summary: Utility scripts based on PyQUDA
 Author-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
 Maintainer-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
@@ -106,6 +106,10 @@ Refer to https://github.com/CLQCD/PyQUDA/wiki/Benchmark for detailed instruction
 
 https://github.com/CLQCD/PyQUDA/wiki/Documentation
 
+## Plugin
+
+https://github.com/CLQCD/PyQUDA/wiki/Plugin
+
 ## Development
 
 We recommend building PyQUDA using in-place mode instead of installing PyQUDA for development.
@@ -113,10 +117,12 @@ We recommend building PyQUDA using in-place mode instead of installing PyQUDA fo
 ```bash
 git clone --recursive https://github.com/CLQCD/PyQUDA.git
 cd PyQUDA
-ln -s pyquda_core/pyquda pyquda
+ln -s pyquda_core/pyquda_comm ./
+ln -s pyquda_core/pyquda ./
 cd pyquda_core
 export QUDA_PATH=/path/to/quda/build/usqcd
 python3 setup.py build_ext --inplace
+cd ..
 ```
 
 Now you can modify Python files in the project and immediately get the new result by running scripts. Adding the root directory to `sys.path` is needed if you are running scripts from other directories.

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/PyQUDA_Utils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyQUDA-Utils
-Version: 0.10.7.dev3
+Version: 0.10.8.dev0
 Summary: Utility scripts based on PyQUDA
 Author-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
 Maintainer-email: SaltyChiang <SaltyChiang@users.noreply.github.com>
@@ -106,6 +106,10 @@ Refer to https://github.com/CLQCD/PyQUDA/wiki/Benchmark for detailed instruction
 
 https://github.com/CLQCD/PyQUDA/wiki/Documentation
 
+## Plugin
+
+https://github.com/CLQCD/PyQUDA/wiki/Plugin
+
 ## Development
 
 We recommend building PyQUDA using in-place mode instead of installing PyQUDA for development.
@@ -113,10 +117,12 @@ We recommend building PyQUDA using in-place mode instead of installing PyQUDA fo
 ```bash
 git clone --recursive https://github.com/CLQCD/PyQUDA.git
 cd PyQUDA
-ln -s pyquda_core/pyquda pyquda
+ln -s pyquda_core/pyquda_comm ./
+ln -s pyquda_core/pyquda ./
 cd pyquda_core
 export QUDA_PATH=/path/to/quda/build/usqcd
 python3 setup.py build_ext --inplace
+cd ..
 ```
 
 Now you can modify Python files in the project and immediately get the new result by running scripts. Adding the root directory to `sys.path` is needed if you are running scripts from other directories.

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/PyQUDA_Utils.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ setup.py
 PyQUDA_Utils.egg-info/PKG-INFO
 PyQUDA_Utils.egg-info/SOURCES.txt
 PyQUDA_Utils.egg-info/dependency_links.txt
+PyQUDA_Utils.egg-info/entry_points.txt
 PyQUDA_Utils.egg-info/requires.txt
 PyQUDA_Utils.egg-info/top_level.txt
 pyquda_io/__init__.py
@@ -29,6 +30,7 @@ pyquda_utils/alg_remez.py
 pyquda_utils/convert.py
 pyquda_utils/core.py
 pyquda_utils/deprecated.py
+pyquda_utils/eigensolve.py
 pyquda_utils/gamma.py
 pyquda_utils/gauge_nd_sun.py
 pyquda_utils/gpt.py

pyquda_utils-0.10.8.dev0/PyQUDA_Utils.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+pyquda_plugins = pyquda_plugins.__main__:main

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/README.md

@@ -60,6 +60,10 @@ Refer to https://github.com/CLQCD/PyQUDA/wiki/Benchmark for detailed instruction
 
 https://github.com/CLQCD/PyQUDA/wiki/Documentation
 
+## Plugin
+
+https://github.com/CLQCD/PyQUDA/wiki/Plugin
+
 ## Development
 
 We recommend building PyQUDA using in-place mode instead of installing PyQUDA for development.
@@ -67,10 +71,12 @@ We recommend building PyQUDA using in-place mode instead of installing PyQUDA fo
 ```bash
 git clone --recursive https://github.com/CLQCD/PyQUDA.git
 cd PyQUDA
-ln -s pyquda_core/pyquda pyquda
+ln -s pyquda_core/pyquda_comm ./
+ln -s pyquda_core/pyquda ./
 cd pyquda_core
 export QUDA_PATH=/path/to/quda/build/usqcd
 python3 setup.py build_ext --inplace
+cd ..
 ```
 
 Now you can modify Python files in the project and immediately get the new result by running scripts. Adding the root directory to `sys.path` is needed if you are running scripts from other directories.

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyproject.toml

@@ -42,3 +42,6 @@ Homepage = "https://github.com/CLQCD/PyQUDA"
 Documentation = "https://github.com/CLQCD/PyQUDA/wiki/Documentation"
 Repository = "https://github.com/CLQCD/PyQUDA.git"
 Issues = "https://github.com/CLQCD/PyQUDA/issues"
+
+[project.scripts]
+pyquda_plugins = "pyquda_plugins.__main__:main"

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_io/_field_utils.py

@@ -1,3 +1,4 @@
+from math import prod
 from typing import List
 
 from mpi4py import MPI
@@ -78,7 +79,7 @@ def gaugePlaquette(latt_size: List[int], gauge: numpy.ndarray):
     plaq[3] = numpy.vdot(gauge[0] @ extended[3, :-1, :-1, :-1, 1:], gauge[3] @ extended[0, 1:, :-1, :-1, :-1]).real
     plaq[4] = numpy.vdot(gauge[1] @ extended[3, :-1, :-1, 1:, :-1], gauge[3] @ extended[1, 1:, :-1, :-1, :-1]).real
     plaq[5] = numpy.vdot(gauge[2] @ extended[3, :-1, 1:, :-1, :-1], gauge[3] @ extended[2, 1:, :-1, :-1, :-1]).real
-    plaq /= int(numpy.prod(latt_size)) * Nc
+    plaq /= prod(latt_size) * Nc
     plaq = getMPIComm().allreduce(plaq, MPI.SUM)
     return numpy.array([plaq.mean(), plaq[:3].mean(), plaq[3:].mean()])
 
@@ -303,7 +304,7 @@ def propagatorToDiracPauli(degrand_rossi: numpy.ndarray):
 #     plaq[3] = numpy.vdot(gauge[1, 1] @ gauge[2, 0], gauge[2, 1] @ gauge[1, 0]).real
 #     plaq[4] = numpy.vdot(gauge[1, 1] @ gauge[3, 0], gauge[3, 1] @ gauge[1, 0]).real
 #     plaq[5] = numpy.vdot(gauge[2, 1] @ gauge[3, 0], gauge[3, 1] @ gauge[2, 0]).real
-#     plaq /= int(numpy.prod(latt_size)) * Nc
+#     plaq /= prod(latt_size) * Nc
 #     plaq = getMPIComm().allreduce(plaq, MPI.SUM)
 #     return numpy.array([plaq.mean(), plaq[:3].mean(), plaq[3:].mean()])
 
@@ -382,6 +383,6 @@ def propagatorToDiracPauli(degrand_rossi: numpy.ndarray):
 #     plaq[3] = numpy.vdot(link_munu[3], link_numu[3]).real
 #     plaq[4] = numpy.vdot(link_munu[4], link_numu[4]).real
 #     plaq[5] = numpy.vdot(link_munu[5], link_numu[5]).real
-#     plaq /= int(numpy.prod(latt_size)) * Nc
+#     plaq /= prod(latt_size) * Nc
 #     plaq = getMPIComm().allreduce(plaq, MPI.SUM)
 #     return numpy.array([plaq.mean(), plaq[:3].mean(), plaq[3:].mean()])

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_plugins/plugin_pyx.py

@@ -205,3 +205,8 @@ def build_plugin_pyx(plugins_root, lib: str, header: str, include_path: str):
         f.write(pylib_pyx)
     with open(os.path.join(plugins_root, f"py{lib}", f"_py{lib}.pyi"), "w") as f:
         f.write(pylib_pyi)
+
+    if os.path.exists(os.path.join(plugins_root, "yacctab.py")):
+        os.remove(os.path.join(plugins_root, "yacctab.py"))
+    if os.path.exists(os.path.join(plugins_root, "lextab.py")):
+        os.remove(os.path.join(plugins_root, "lextab.py"))

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_plugins/pygwu/__init__.py

@@ -5,7 +5,7 @@ from typing import Sequence
 import numpy as np
 
 from pyquda_comm import initGrid, initDevice, getLogger, setGridMap, readMPIFile, readMPIFileInChunks
-from pyquda_comm.field import LatticeInfo, LatticeGauge, LatticeFermion, MultiLatticeFermion, evenodd
+from pyquda_comm.field import evenodd, LatticeInfo, LatticeGauge, LatticeFermion, MultiLatticeFermion
 from . import _pygwu as gwu
 
 setGridMap("TZYX_FASTEST")

pyquda_utils-0.10.8.dev0/pyquda_utils/_version.py

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

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/convert.py

@@ -1,7 +1,7 @@
 from typing import Union
 
 from pyquda import getLogger
-from pyquda_comm.field import (
+from pyquda.field import (
     LatticeLink,
     LatticeFermion,
     LatticePropagator,

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/core.py

@@ -1,3 +1,4 @@
+from math import prod
 from typing import List, Literal, Union
 
 import numpy
@@ -17,11 +18,13 @@ from pyquda import (
     setDefaultLattice,
     getDefaultLattice,
     getCUDABackend,
+    getCUDADevice,
+    getCUDAComputeCapability,
     getLogger,
     setLoggerLevel,
     dirac as fermion,
 )
-from pyquda_comm.field import (
+from pyquda.field import (  # noqa: F401
     Ns,
     Nc,
     Nd,
@@ -238,8 +241,8 @@ def gatherLattice(data: numpy.ndarray, axes: List[int], reduce_op: Literal["sum"
     reduce_axis = tuple([keep[0] + d for d in range(4) if axes[d] == -1])
     prefix = data.shape[: keep[0]]
     suffix = data.shape[keep[-1] + 1 :]
-    prefix_size = int(numpy.prod(prefix))
-    suffix_size = int(numpy.prod(suffix))
+    prefix_size = prod(prefix)
+    suffix_size = prod(suffix)
 
     if getMPIRank() == root:
         sendbuf = numpy.ascontiguousarray(data.reshape(-1))

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/deprecated.py

@@ -1,7 +1,7 @@
 from typing import List
 
 from pyquda import getLogger, getGridSize, pyquda as quda, enum_quda
-from pyquda_comm.field import LatticeFermion, LatticeGauge, LatticeInfo, LatticePropagator, Nc, Ns
+from pyquda.field import LatticeFermion, LatticeGauge, LatticeInfo, LatticePropagator, Nc, Ns
 from pyquda.dirac.abstract import FermionDirac
 
 

pyquda_utils-0.10.8.dev0/pyquda_utils/eigensolve.py

@@ -0,0 +1,43 @@
+from pyquda import pyquda as quda
+from pyquda.enum_quda import QudaDslashType, QudaBoolean, QudaEigType, QudaEigSpectrumType
+from pyquda.field import LatticeGauge, MultiLatticeStaggeredFermion
+
+
+def laplace3d(
+    gauge: LatticeGauge, n_ev: int, n_kr: int, tol: int, max_restarts: int, poly_deg: int = 1, poly_cut: float = 0
+):
+    import numpy
+
+    latt_info = gauge.latt_info
+    gauge.gauge_dirac.invert_param.dslash_type = QudaDslashType.QUDA_LAPLACE_DSLASH
+    gauge.gauge_dirac.invert_param.mass = -1
+    gauge.gauge_dirac.invert_param.kappa = 1 / 6
+    gauge.gauge_dirac.invert_param.laplace3D = 3
+    gauge.gauge_dirac.loadGauge(gauge)
+
+    eig_param = quda.QudaEigParam()
+    eig_param.invert_param = gauge.gauge_dirac.invert_param
+    eig_param.eig_type = QudaEigType.QUDA_EIG_TR_LANCZOS_3D
+    eig_param.use_dagger = QudaBoolean.QUDA_BOOLEAN_FALSE
+    eig_param.use_norm_op = QudaBoolean.QUDA_BOOLEAN_FALSE
+    eig_param.use_pc = QudaBoolean.QUDA_BOOLEAN_FALSE
+    eig_param.compute_gamma5 = QudaBoolean.QUDA_BOOLEAN_FALSE
+    eig_param.spectrum = QudaEigSpectrumType.QUDA_SPECTRUM_SR_EIG
+    eig_param.n_ev = n_ev
+    eig_param.n_kr = n_kr
+    eig_param.n_conv = n_ev
+    eig_param.tol = tol
+    eig_param.ortho_dim = 3
+    eig_param.ortho_dim_size_local = latt_info.Lt
+    eig_param.vec_infile = b""
+    eig_param.vec_outfile = b""
+    eig_param.max_restarts = max_restarts
+    eig_param.use_poly_acc = QudaBoolean(poly_deg > 1)
+    eig_param.poly_deg = poly_deg
+    eig_param.a_min = poly_cut
+    eig_param.a_max = 2
+
+    evals = numpy.empty((latt_info.Lt * n_ev), "<c16")
+    evecs = MultiLatticeStaggeredFermion(latt_info, n_ev)
+    quda.eigensolveQuda(evecs.data_ptrs, evals, eig_param)
+    return evals.reshape(latt_info.Lt, n_ev).transpose(1, 0).real, evecs

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/hmc_param.py

@@ -6,7 +6,7 @@ from typing import Literal, Sequence, Tuple
 from gmpy2 import mpfr
 
 from pyquda_comm import getMPIComm, getMPIRank
-from pyquda_comm.field import X, Y, Z, T
+from pyquda.field import X, Y, Z, T
 from pyquda.action.abstract import LoopParam, RationalParam
 from .alg_remez import AlgRemez
 

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/io/__init__.py

@@ -2,7 +2,7 @@ from typing import List, Union
 
 import numpy
 
-from pyquda_comm.field import (
+from pyquda.field import (
     Ns,
     Nc,
     LatticeInfo,

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/milc_rhmc_param.py

@@ -1,7 +1,7 @@
 from math import log
 from typing import Dict, Tuple
 
-from pyquda_comm.field import X, Y, Z, T
+from pyquda.field import X, Y, Z, T
 from pyquda.action.abstract import LoopParam, RationalParam
 
 

{pyquda_utils-0.10.7.dev3 → pyquda_utils-0.10.8.dev0}/pyquda_utils/source.py

@@ -1,7 +1,7 @@
 from typing import List, Literal, Union
 
 from pyquda import getGridSize, getLogger
-from pyquda_comm.field import (
+from pyquda.field import (
     Ns,
     Nc,
     LatticeInfo,

pyquda_utils-0.10.7.dev3/pyquda_utils/_version.py

@@ -1 +0,0 @@
-__version__ = "0.10.7.dev3"