Use QUDA for lattice QCD calculation with Python

Jiang, Xiangyu; Shi, Chunjiang; Chen, Ying; Gong, Ming; Yang, Yi-Bo

Use QUDA for lattice QCD calculation with Python

Xiangyu Jiang
(
- Beijing, Inst. High Energy Phys. and
- Beijing, Inst. Theor. Phys.
)
,
Chunjiang Shi
(
- Beijing, Inst. High Energy Phys.
)
,
Ying Chen
(
- Beijing, Inst. High Energy Phys.
)
,
Ming Gong
(
- Beijing, Inst. High Energy Phys.
)
,
Yi-Bo Yang
(
- Beijing, Inst. Theor. Phys. and
- UCAS, Beijing and
- HIAS, UCAS, Hangzhou and
- ICTP-AP, Beijing
)

Nov 13, 2024

11 pages

e-Print:

2411.08461 [hep-lat]

View in:

ADS Abstract Service

pdf

reference search3 citations

Citations per year

Abstract: (arXiv)

We developed PyQUDA, a Python wrapper for QUDA written in Cython, designed to facilitate lattice QCD calculations using the Python programming language. PyQUDA leverages the optimized linear algebra capabilities of NumPy/CuPy/PyTorch, along with the highly optimized lattice QCD operations provided by QUDA to accelerate research. This integration simplifies the process of writing calculation codes, enabling researchers to build more complex Python packages like EasyDistillation for specific physics objectives. PyQUDA supports a range of lattice QCD operations, including hybrid Monte Carlo (HMC) with N-flavor clover/HISQ fermions and inversion for the Wilson/clover/HISQ fermion action with the multigrid solver. It also includes utility functions for reading lattice QCD data stored in Chroma, MILC, and

\chi

QCD formats. Type hints are supported by stub files and multi-GPU support is provided through mpi4py.

Note:

11 pages, 3 listings

References(45)

Figures(0)

[1]

Solving Lattice QCD systems of equations using mixed precision solvers on GPUs

QUDA

Collaboration

•

M.A. Clark
(
- Harvard-Smithsonian Ctr. Astrophys. and
- Harvard U.
)

et al.

- Comput.Phys.Commun. 181 (2010) 1517-1528
•
e-Print:
- 0911.3191
•
DOI:
- 10.1016/j.cpc.2010.05.002

[2]

Scaling lattice QCD beyond 100 GPUs

QUDA

Collaboration

•

R. Babich
(
- Boston U., Ctr. Comp. Sci.
)

et al.

•
e-Print:
- 1109.2935
•
DOI:
- 10.1145/2063384.2063478

[3]

The Chroma software system for lattice QCD

SciDAC and
LHPC and
UKQCD

Collaborations

•

Robert G. Edwards
(
- Jefferson Lab
)

et al.

- Nucl.Phys.B Proc.Suppl. 140 (2005) 832,
•
e-Print:
- hep-lat/0409003
•
DOI:
- 10.1016/j.nuclphysbps.2004.11.254

[4]

A Framework for Lattice QCD Calculations on GPUs

e-Print:
- 1408.5925
•
DOI:
- 10.1109/IPDPS.2014.112

[5]

Array programming with NumPy

Charles R. Harris
(
- Unlisted, US, UT
)
,
K. Jarrod Millman
(
- UC, Berkeley and
- LBL, Berkeley
)
,
Stéfan J. van der Walt
(
- Stellenbosch U. and
- UC, Berkeley
)
,
Ralf Gommers
(
- Texas U.
)
,
Pauli Virtanen
(
- Jyvaskyla U.
)

et al.

- Nature 585 (2020) 7825, 357-362
•
e-Print:
- 2006.10256
•
DOI:
- 10.1038/s41586-020-2649-2

[6]

Cupy: A numpy-compatible library for nvidia gpu calculations, in Proceedings of Workshop on Machine Learning Systems (LearningSys) in The Thirty-first Annual Conference on Neural Information Processing Systems (NIPS)

R. Okuta
,
Y. Unno
,
D. Nishino
,
S. Hido
,
C. Loomis

[7]

PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation, in 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2 (ASPLOS ’24) (ACM,)

J. Ansel
,
E. Yang
,
H. He
,
N. Gimelshein
,
A. Jain

et al.

[8]

Cython: The Best of Both Worlds

et al.

- Comput.Sci.Eng. 13 (2011) 2, 31-39
•
DOI:
- 10.1109/MCSE.2010.118

[9]

PEP 484 - Type Hints

- https://peps.python.org/pep0484/

[10]

Official page for Language Server Protocol, https:// microsoft.github.io/language-server-protocol/

[11]

Lyncs-API: a Python API for Lattice QCD applications

Simone Bacchio
(
- Cyprus Inst.
)
,
Jacob Finkenrath
(
- Cyprus Inst.
)
,
Christodoulos Stylianou
(
- Cyprus Inst.
)

- PoS LATTICE2021 (2022) 542
•
e-Print:
- 2201.03873
•
DOI:
- 10.22323/1.396.0542

[12]

Running HMC Simulation with Python via QUDA

- PoS LATTICE2022 (2023) 346
•
e-Print:
- 2212.06657
•
DOI:
- 10.22323/1.430.0346

[13]

High-performance pythonc++ bindings with pypy and cling, in6th Workshop on Python for High-Performance and Scientific Computing (PyHPC) pp. 27-35

W.T. Lavrijsen
,
A. Dutta

[14]

Benchmarks - nanobind documentation, https:// nanobind.readthedocs.io/en/latest/benchmark.html

[15]

lehner/gpt:

A. Python

[16]

Grid: A next generation data parallel C++ QCD library

e-Print:
- 1512.03487

[17]

eliben/pycparser:snake: Complete C99 parser in pure Python

- https://github.com/eliben/pycparser

[18]

mpi4py: Status Update After 12 Years of Development

L. Dalcin
,
Y.-L.L. Fang

- Comput.Sci.Eng. 23 (2021) 47

[19]

Improved Continuum Limit Lattice Action for QCD with Wilson Fermions

B. Sheikholeslami
(
- Hamburg U.
)
,
R. Wohlert
(
- Hamburg U.
)

- Nucl.Phys.B 259 (1985) 572,
- Nucl. Phys. B259 ( 1985) 572-596 and Hamburg Desy - DESY 85-024 (85,REC.MAR.) 26p
•
DOI:
- 10.1016/0550-3213(85)90002-1

[20]

Nonperturbative improvement of the anisotropic Wilson QCD action

Timothy R. Klassen
(
- Florida State U., SCRI
)

- Nucl.Phys.B Proc.Suppl. 73 (1999) 918-920,
•
e-Print:
- hep-lat/9809174
•
DOI:
- 10.1016/S0920-5632(99)85244-X

[21]

Accelerating lattice QCD multigrid on GPUs using fine-grained parallelization

QUDA

Collaboration

•

M.A. Clark
(
- NVIDIA, Santa Clara
)

et al.

- ,
- Proceedings of the International Conference for High Performance Computing, Networking, Storage and Analysis (SC '16), Article 68 (November, 2016)
•
e-Print:
- 1612.07873
•
DOI:
- 10.5555/3014904.3014995

[22]

The Rational Hybrid Monte Carlo Algorithm

M.A. Clark
(
- Boston U., Ctr. Comp. Sci.
)

- PoS LAT2006 (2006) 004
•
e-Print:
- hep-lat/0610048
•
DOI:
- 10.22323/1.032.0004

[23]

Continuum Limit Improved Lattice Action for Pure Yang-Mills Theory. 1.

P. Weisz
(
- Hamburg U.
)

- Nucl.Phys.B 212 (1983) 1-17
•
DOI:
- 10.1016/0550-3213(83)90595-3

[24]

On the viability of lattice perturbation theory

G.Peter Lepage
(
- Cornell U., LNS
)
,
Paul B. Mackenzie
(
- Fermilab
)

- Phys.Rev.D 48 (1993) 2250-2264
•
e-Print:
- hep-lat/9209022
•
DOI:
- 10.1103/PhysRevD.48.2250

[25]

milc-qcd/milc qcd: MILC collaboration code for lattice QCD calculations

- https://github.com/milc-qcd/

1-25 of 45
1
2
25 / page