WarpX Code Shines at the Exascale Level

February 1, 2023

ECP-funded project takes plasma accelerator modeling to new heights.

Feb. 1, 2023 — Long valued for their role in scientific discovery and in a variety of medical and industrial applications, particle accelerators have been used in many areas of fundamental research and credited with enabling Nobel Prize–winning research in physics and chemistry. But these high-end instruments also occupy a lot of space and carry hefty price tags. Even smaller accelerators, such as those used in medical centers for proton therapy, need large spaces to accommodate their hardware, power supplies, and radiation shielding.

Fortunately, over the last several years physicists, engineers, and computational scientists have been working to create more affordable and accessible particle accelerators by shrinking both the size and the cost while increasing the capability. One of the most exciting developments in these efforts is the plasma accelerator, which uses lasers or particle beams rather than radio-frequency waves to generate the accelerating field, allowing these devices to support accelerating-electric fields many orders of magnitude greater than conventional accelerators with a much smaller footprint—even able to fit on a tabletop.

The reduced size of plasma accelerators, however, presents challenges in controlling the intricate ultrafast processes at play, which are often on the picosecond and micrometer scale. Thus, realizing their compact designs requires novel mathematical and software capabilities to enable high-performance, high-fidelity modeling that can capture the full complexity of acceleration processes over a large range of space and timescales—simulations that are often computationally intensive.

To address this issue, the WarpX project—a Lawrence Berkeley National Laboratory (Berkeley Lab)–led effort that also drew the attention and support of DOE’s Exascale Computing Project (ECP)—has spent the last 6 years creating a novel, highly parallel and highly optimized single-source simulation code for modeling plasma-based particle colliders on cutting-edge exascale supercomputers, with broad importance for other accelerators and related problems.

WarpX enables computational explorations of key physics questions in the transport and acceleration of particle beams in long chains of plasma channels, which could yield significant savings in the design and characterization of plasma-based colliders before they are built. Using exascale modeling to validate these devices could also lead to broader applications, such as sterilizing food or toxic waste, implanting ions in semiconductors, treating cancer, advancing fusion research, and developing new drugs.

“A lot of research still needs to be done and, like most things with plasma, you need big simulation tools because it’s complicated due to the large number of space and time scales. This is where WarpX comes into play,” said Jean-Luc Vay, a senior scientist at Berkeley Lab who heads the Lab’s Accelerator Modeling Program in the Applied Physics and Accelerator Technologies division, leads the WarpX development team at Berkeley Lab, and is co-PI (along with Berkeley Lab’s Ann Almgren) for the ECP WarpX development team. Outside of Berkeley Lab, the ECP WarpX development team includes collaborators from Lawrence Livermore National Laboratory and SLAC National Accelerator Laboratory.

Next-Generation PIC Solution

“The idea of WarpX is to provide a particle-in-cell (PIC) solution [a technique used to solve a class of plasma physics problems from first principles] that can tackle problems that are much bigger and faster than we could do before, allowing us to explore solutions for plasma acceleration at scales not previously possible,” Vay said. “We don’t need to do just simulations—we also need to study tolerances, which is highly computationally demanding. So we are pushing the state of the art with high performance computing.” In addition, since the PIC method is important for modeling many other beams and plasma physics problems, “the code is being designed to be generic and will advance solutions for a wide range of problems in accelerators, fusion, and more, far beyond the main focus on plasma-based colliders of the ECP WarpX project,” Vay added.

The original version of the code—dubbed simply “Warp” and developed by Alex Friedman and collaborators (including Vay and other WarpX developers) for DOE’s Fusion Energy Science program—used Fortran subroutines for fast number crunching that were wrapped in a Python top layer for simulations control and steering. While that version of the code incorporated many original, cutting-edge algorithms, like adaptive mesh refinement (AMR), it was not easily portable to run efficiently on both CPUs and GPUs; this hybrid architecture is a hallmark of emerging exascale supercomputers. As a result, “it became quickly clear that a full rewrite of the code using portable C++ primitives was the way to go for developing a single source code that runs efficiently on both platforms,” Vay said—a decision that led to one of the biggest challenges the WarpX team has faced so far.

So the team turned to AMReX, an ECP software library that provides a robust, efficient, and scalable implementation of AMR capabilities and versatile portable C++ primitives, which was developed by a team led by John Bell and Ann Almgren in the Applied Mathematics and Computational Research Division (AMCRD) at Berkeley Lab.

WarpX: longitudinal electric field in a laser-plasma accelerator rendered with the ECP software libraries Ascent & VTK-m as the simulation was running. Credit: Axel Huebl, Berkeley Lab.

“AMReX is a numerical library that helps us implement physical block-structured mesh refinement algorithms on multiple computer architectures,” said Axel Huebl, a computational physicist and research software engineer in Berkeley Lab’s Accelerator Technology and Applied Physics Division and lead developer of WarpX. “Mesh refinement enables us to focus the computational power on the most interesting parts of a simulation, while staying effective for the larger, macroscopic evolution of, for instance, the plasma physics we model.”

Toward this end, as the WarpX team simulated longer chains of plasma accelerators at higher grid resolutions, the efficiency of the simulations was limited by the numerical limitations of the existing algorithms. So new algorithms were devised that removed some specific limitations, resulting in speedups of an order of magnitude or more in some cases. And after just one year of development, the AMReX team had the enhanced WarpX code up and running on GPUs, Vay noted.

“In addition to the question of what languages and programming tools we should use, we also had to change the way a lot of the algorithms are implemented,” said Andrew Myers, a computer systems engineer in AMCRD and a member of the WarpX team who has been instrumental in the AMReX implementation. “For example, we had to redesign the way a lot of the particle algorithms in AMReX work.”

“The AMReX team did an amazing job, doing a lot of tests to determine what the solutions could be and then developing this layer that provides a single source code that allows us to compile for CPUs or NVIDIA, AMD, or Intel GPUs,” Vay said.

Four Supercomputers, a Gordon Bell Prize, and More

With support from ECP and team members’ respective institutions, these development efforts are already paying off. In simulations run in 2022, the WarpX project demonstrated a 500x improvement in performance on the exascale supercomputer Frontier over the preceding version of Warp and was the first ECP application to reach its project goal by running at scale on Frontier. It was also the first code to perform 3D simulations of laser-matter interactions on Frontier, Fugaku, and Summit—something that has so far been out of reach for standard codes.

“With WarpX we can take the same code base and compile it for all of those different machines without having to make any changes to the code, and that was a big focus point of the effort and the project,” Myers said.

“That is something that ECP really puts an emphasis on,” Huebl added. “If you are working for such a long time on a project, how do you make this development sustainable? Our codes outlive the machines most of the time, so we intentionally plan it in a way that we can continue in five years when the next machine is coming along, and we don’t have to start from scratch and can optimize the code without having to rewrite everything.”

Another key milestone for the WarpX project was the successful implementation and deployment of the code for first-of-kind mesh-refined massively parallel 3D PIC simulations of kinetic plasma optimized on 4 of the 10 fastest supercomputers in the world (Frontier, Fugaku, Summit, and Perlmutter). This accomplishment earned the WarpX development team and collaborators from France and Japan the Association for Computing Machinery’s prestigious Gordon Bell Prize in 2022.

And the WarpX project is far from over. Several activities are already under way that will further leverage the latest iteration of the code for a variety of applications:

  • WarpX is being applied to the exploration of outstanding questions for plasma-based collider designs on tens of consecutive plasma stages, toward the modeling of multi-TeV high-energy physics colliders based on tens to thousands of plasma-based accelerator stages. Novel algorithms are being studied to improve the accuracy and speed of the code for these studies, including a more versatile and robust implementation of AMR.
  • A growing number of users in research labs, academia, and industry are applying WarpX to topics such as laser-ion acceleration, structure-based wakefield acceleration, laser-plasma interaction, plasma instabilities, plasma mirrors, fusion devices, magnetic fusion sheaths, magnetic reconnection, pulsars physics, thermionic converters, electron clouds in accelerators. Several of them are also contributing to the code testing and development.
  • The success of WarpX has prompted spinoff projects such as ImpactX, a rewrite of the popular conventional accelerator suite IMPACT (now ImpactX), and HiPACE++, a rewrite of the quasistatic code HiPACE for plasma accelerators. Both have been rewritten for CPUs and GPUs using the AMReX library and sharing data structures and modules with WarpX. Another spinoff is Artemis, which is built on top of WarpX with additional functionalities for the modeling of micromagnetics and electrodynamic waves in next-generation microelectronics.

“WarpX was built as a general code with one target application in mind, but it can do much more, and people are already exploring further applications,” Vay said.

Being part of the ECP effort has had a demonstrable impact on this body of research, he added. “The ECP project has been an incredible boost to our scientific code developments and productivity,” he said. “Many of us on the WarpX team agree that it has fostered, between WarpX, AMReX, and partners, the best collaboration experience that we have had to date in our scientific lives. In addition, ECP is enabling us to realize our vision of an integrated PIC ecosystem of codes that is part of a community of integrated scientific software.”

Beyond ECP, more work is now needed to further boost the speed and efficiency of the code, and to address the broad applications in accelerators, fusion, and more. “Make no mistake,” said Vay, “the work on WarpX is not over.” For example, continued work on novel algorithms is needed to improve the accuracy and speed of the code, including a more versatile and robust implementation of AMR.

The ECP WarpX Development Team

Berkeley Lab

  • Ann Almgren (co-PI)
  • Arianna Formenti
  • Marco Garten
  • Kevin Gott
  • Junmin Gu
  • Axel Huebl
  • Revathi Jambunathan
  • Hannah Klion
  • Prabhat Kumar
  • Rémi Lehe
  • Andrew Myers
  • Ryan Sandberg
  • Olga Shapoval
  • Maxence Thevenet (now DESY)
  • Jean-Luc Vay (PI)
  • Weiqun Zhang
  • Edoardo Zoni

Lawrence Livermore National Laboratory

  • David Grote

SLAC

  • Lixin Ge
  • Cho Ng

(Plus contributors from laboratories, universities and industry organizations in the United States, Europe, and Asia)


Source: Kathy Kincade, Lawrence Berkeley National Laboratory and ECP

Subscribe to HPCwire's Weekly Update!

Be the most informed person in the room! Stay ahead of the tech trends with industy updates delivered to you every week!

Pegasus ‘Big Memory’ Supercomputer Now Deployed at the University of Tsukuba

March 25, 2023

In the bevy of news from Nvidia's GPU Technology Conference this week, another new system has come to light: Pegasus, which entered operations at the University of Tsukuba’s Center for Computational Sciences in January Read more…

EuroHPC Summit: Tackling Exascale, Energy, Industry & Sovereignty

March 24, 2023

As the 2023 EuroHPC Summit opened in Gothenburg on Monday, Herbert Zeisel – chair of EuroHPC’s Governing Board – commented that the undertaking had “left its teenage years behind.” Indeed, a sense of general ma Read more…

Is Fortran the Best Programming Language? Asking ChatGPT

March 23, 2023

I recently wrote about my experience with interviewing ChatGPT here. As promised, in this follow-on and conclusion of my interview, I focus on Fortran and other languages. All in good fun. I hope you enjoy the conclusion of my interview. After my programming language questions, I conclude with a few notes... Read more…

Nvidia Doubling Down on China Market in the Face of Tightened US Export Controls

March 23, 2023

Chipmakers are tightlipped on China activities following a U.S. crackdown on hardware exports to the country. But Nvidia remains unfazed, and is doubling down on China being an important country for its computing hardwar Read more…

Intel’s Sapphire Rapids Comes to Australia’s Gadi Supercomputer

March 22, 2023

Until the launch of Pawsey’s Setonix system last year, NCI’s Gadi system – launched in 2020 – was Australia’s most powerful publicly ranked supercomputer. Now, the system has received a major boost powered by I Read more…

AWS Solution Channel

Shutterstock_2206622211

Install optimized software with Spack configs for AWS ParallelCluster

With AWS ParallelCluster, you can choose a computing architecture that best matches your HPC application. But, HPC applications are complex. That means they can be challenging to get working well. Read more…

 

Get the latest on AI innovation at NVIDIA GTC

Join Microsoft at NVIDIA GTC, a free online global technology conference, March 20 – 23 to learn how organizations of any size can power AI innovation with purpose-built cloud infrastructure from Microsoft. Read more…

Nvidia Announces BlueField-3 GA, Oracle Cloud Is Early User

March 21, 2023

Nvidia today announced general availability for its BlueField-3 data processing unit (DPU) along with impressive early deployments including Oracle Cloud Infrastructure. First described in 2021 and now being delivered, B Read more…

Pegasus ‘Big Memory’ Supercomputer Now Deployed at the University of Tsukuba

March 25, 2023

In the bevy of news from Nvidia's GPU Technology Conference this week, another new system has come to light: Pegasus, which entered operations at the University Read more…

EuroHPC Summit: Tackling Exascale, Energy, Industry & Sovereignty

March 24, 2023

As the 2023 EuroHPC Summit opened in Gothenburg on Monday, Herbert Zeisel – chair of EuroHPC’s Governing Board – commented that the undertaking had “lef Read more…

Nvidia Doubling Down on China Market in the Face of Tightened US Export Controls

March 23, 2023

Chipmakers are tightlipped on China activities following a U.S. crackdown on hardware exports to the country. But Nvidia remains unfazed, and is doubling down o Read more…

Nvidia Announces BlueField-3 GA, Oracle Cloud Is Early User

March 21, 2023

Nvidia today announced general availability for its BlueField-3 data processing unit (DPU) along with impressive early deployments including Oracle Cloud Infras Read more…

Nvidia Announces ‘Tokyo-1’ Generative AI Supercomputer Amid Gradual H100 Rollout

March 21, 2023

Nvidia’s Hopper-generation H100 GPU is continuing its slow march toward “current-generation.” After Nvidia announced that the H100 was in “full producti Read more…

DGX Cloud Is Here: Nvidia’s AI Factory Services Start at $37,000

March 21, 2023

If you are a die-hard Nvidia loyalist, be ready to pay a fortune to use its AI factories in the cloud. Renting the GPU company's DGX Cloud, which is an all-inclusive AI supercomputer in the cloud, starts at $36,999 per instance for a month. The rental includes access to a cloud computer with eight Nvidia H100 or A100 GPUs and 640GB... Read more…

Quantum Bits: IBM-Cleveland Clinic Launch; D-Wave Adds Solver; DOE/AWS Offer QICK

March 20, 2023

IBM today launched the first installation of an IBM Quantum System One at a collaborator site in the U.S. – this one is at the Cleveland Clinic where IBM’s Read more…

SCA23: Pawsey’s Mark Stickells on Sustainable Australian Supercomputing

March 17, 2023

“While the need for supercomputing is great, we have, in my view, reached a tipping point,” said Mark Stickells, executive director of Australia’s Pawsey Read more…

CORNELL I-WAY DEMONSTRATION PITS PARASITE AGAINST VICTIM

October 6, 1995

Ithaca, NY --Visitors to this year's Supercomputing '95 (SC'95) conference will witness a life-and-death struggle between parasite and victim, using virtual Read more…

SGI POWERS VIRTUAL OPERATING ROOM USED IN SURGEON TRAINING

October 6, 1995

Surgery simulations to date have largely been created through the development of dedicated applications requiring considerable programming and computer graphi Read more…

U.S. Will Relax Export Restrictions on Supercomputers

October 6, 1995

New York, NY -- U.S. President Bill Clinton has announced that he will definitely relax restrictions on exports of high-performance computers, giving a boost Read more…

Dutch HPC Center Will Have 20 GFlop, 76-Node SP2 Online by 1996

October 6, 1995

Amsterdam, the Netherlands -- SARA, (Stichting Academisch Rekencentrum Amsterdam), Academic Computing Services of Amsterdam recently announced that it has pur Read more…

Cray Delivers J916 Compact Supercomputer to Solvay Chemical

October 6, 1995

Eagan, Minn. -- Cray Research Inc. has delivered a Cray J916 low-cost compact supercomputer and Cray's UniChem client/server computational chemistry software Read more…

NEC Laboratory Reviews First Year of Cooperative Projects

October 6, 1995

Sankt Augustin, Germany -- NEC C&C (Computers and Communication) Research Laboratory at the GMD Technopark has wrapped up its first year of operation. Read more…

Sun and Sybase Say SQL Server 11 Benchmarks at 4544.60 tpmC

October 6, 1995

Mountain View, Calif. -- Sun Microsystems, Inc. and Sybase, Inc. recently announced the first benchmark results for SQL Server 11. The result represents a n Read more…

New Study Says Parallel Processing Market Will Reach $14B in 1999

October 6, 1995

Mountain View, Calif. -- A study by the Palo Alto Management Group (PAMG) indicates the market for parallel processing systems will increase at more than 4 Read more…

Leading Solution Providers

Contributors

CORNELL I-WAY DEMONSTRATION PITS PARASITE AGAINST VICTIM

October 6, 1995

Ithaca, NY --Visitors to this year's Supercomputing '95 (SC'95) conference will witness a life-and-death struggle between parasite and victim, using virtual Read more…

SGI POWERS VIRTUAL OPERATING ROOM USED IN SURGEON TRAINING

October 6, 1995

Surgery simulations to date have largely been created through the development of dedicated applications requiring considerable programming and computer graphi Read more…

U.S. Will Relax Export Restrictions on Supercomputers

October 6, 1995

New York, NY -- U.S. President Bill Clinton has announced that he will definitely relax restrictions on exports of high-performance computers, giving a boost Read more…

Dutch HPC Center Will Have 20 GFlop, 76-Node SP2 Online by 1996

October 6, 1995

Amsterdam, the Netherlands -- SARA, (Stichting Academisch Rekencentrum Amsterdam), Academic Computing Services of Amsterdam recently announced that it has pur Read more…

Cray Delivers J916 Compact Supercomputer to Solvay Chemical

October 6, 1995

Eagan, Minn. -- Cray Research Inc. has delivered a Cray J916 low-cost compact supercomputer and Cray's UniChem client/server computational chemistry software Read more…

NEC Laboratory Reviews First Year of Cooperative Projects

October 6, 1995

Sankt Augustin, Germany -- NEC C&C (Computers and Communication) Research Laboratory at the GMD Technopark has wrapped up its first year of operation. Read more…

Sun and Sybase Say SQL Server 11 Benchmarks at 4544.60 tpmC

October 6, 1995

Mountain View, Calif. -- Sun Microsystems, Inc. and Sybase, Inc. recently announced the first benchmark results for SQL Server 11. The result represents a n Read more…

New Study Says Parallel Processing Market Will Reach $14B in 1999

October 6, 1995

Mountain View, Calif. -- A study by the Palo Alto Management Group (PAMG) indicates the market for parallel processing systems will increase at more than 4 Read more…

SC22 Booth Videos

AMD @ SC22
Altair @ SC22
AWS @ SC22
Ayar Labs @ SC22
CoolIT @ SC22
Cornelis Networks @ SC22
DDN @ SC22
Dell Technologies @ SC22
HPE @ SC22
Intel @ SC22
Intelligent Light @ SC22
Lancium @ SC22
Lenovo @ SC22
Microsoft and NVIDIA @ SC22
One Stop Systems @ SC22
Penguin Solutions @ SC22
QCT @ SC22
Supermicro @ SC22
Tuxera @ SC22
Tyan Computer @ SC22
  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire