Multiple right hand side multigrid for domain wall fermions with a multigrid preconditioned block conjugate gradient algorithm

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteur principal: Boyle, Peter A
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914940041822208
author Boyle, Peter A
author_facet Boyle, Peter A
contents We introduce a class of efficient multiple right-hand side multigrid algorithm for domain wall fermions. The simultaneous solution for a modest number of right hand sides concurrently allows for a significant reduction in the time spent solving the coarse grid operator in a multigrid preconditioner. We introduce a preconditioned block conjuate gradient with a multigrid preconditioner, giving additional algorithmic benefit from the multiple right hand sides. There is also a very significant additional to computation rate benefit to multiple right hand sides. This both increases the arithmetic intensity in the coarse space and increases the amount of work being performed in each subroutine call, leading to excellent performance on modern GPU architectures. Further, the software implementation makes use of vendor linear algebra routines (batched GEMM) that can make use of high throughput tensor hardware on recent Nvidia, AMD and Intel GPUs. The cost of the coarse space is made sub-dominant in this algorithm, and benchmarks from the Frontier supercomputer system show up to a factor of twenty speed up over the standard red-black preconditioned conjugate gradient algorithm on a large system with physical quark masses.
format Preprint
id arxiv_https___arxiv_org_abs_2409_03904
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Multiple right hand side multigrid for domain wall fermions with a multigrid preconditioned block conjugate gradient algorithm
Boyle, Peter A
High Energy Physics - Lattice
Distributed, Parallel, and Cluster Computing
Numerical Analysis
We introduce a class of efficient multiple right-hand side multigrid algorithm for domain wall fermions. The simultaneous solution for a modest number of right hand sides concurrently allows for a significant reduction in the time spent solving the coarse grid operator in a multigrid preconditioner. We introduce a preconditioned block conjuate gradient with a multigrid preconditioner, giving additional algorithmic benefit from the multiple right hand sides. There is also a very significant additional to computation rate benefit to multiple right hand sides. This both increases the arithmetic intensity in the coarse space and increases the amount of work being performed in each subroutine call, leading to excellent performance on modern GPU architectures. Further, the software implementation makes use of vendor linear algebra routines (batched GEMM) that can make use of high throughput tensor hardware on recent Nvidia, AMD and Intel GPUs. The cost of the coarse space is made sub-dominant in this algorithm, and benchmarks from the Frontier supercomputer system show up to a factor of twenty speed up over the standard red-black preconditioned conjugate gradient algorithm on a large system with physical quark masses.
title Multiple right hand side multigrid for domain wall fermions with a multigrid preconditioned block conjugate gradient algorithm
topic High Energy Physics - Lattice
Distributed, Parallel, and Cluster Computing
Numerical Analysis
url https://arxiv.org/abs/2409.03904