NWChemEx: Computational Chemistry Code for the Exascale Era

By Linda Barney

April 29, 2021

A team working on biofuel research is rewriting the decades-old NWChem software program for the exascale era. The new software, NWChemEx, will enable computational chemistry software to effectively run on exascale supercomputers. The team, part of Argonne National Laboratory’s Early Science Program (ESP), estimates that the future Argonne Aurora supercomputer, along with the new NWChemEx algorithms, will provide up to a 1,000x increase in the size of the chemical systems that can be studied.

Distinguished Professor Theresa Windus, Ames Laboratory and Iowa State University, is the principal investigator on the project. Windus states, “One of the goals of our research is to find more accurate information about chemical systems to aid in developing new methods for converting biomass into biofuels. When the Aurora supercomputer comes online at Argonne, it will be one of the largest supercomputers available. Running computer simulations of our biofuel research on Aurora using the NWChemEx software will allow the team to investigate larger molecular systems and perform deeper exploration of the energy landscape for molecules and materials. That additional knowledge provides a better understanding of the fundamental chemical reactions and their properties.”

In addition to Windus, the research team involves a broad spectrum of expertise located at Ames Laboratory, Argonne National Laboratory, Brookhaven National Laboratory, Lawrence Berkeley National Laboratory, Oak Ridge National Laboratory, Pacific Northwest National Laboratory, and Virginia Tech, and frequent collaborations with Intel, AMD, HPE and Nvidia.

Updating NWChem code for Exascale computing

NWChem is a widely used open source computational chemistry package that includes both quantum chemical and molecular dynamics functionality. However, the NWChem code was designed for computer architectures that no longer exist. NWChemEx is a complete rewrite of the software – it’s modularized and flexible from its initial design and uses the power of C++ – enabling the team to move away from the megalithic design pattern that’s plagued quantum chemistry since the 70s. A primary focus of NWChemEx is designing the code so it runs on graphic processing units (GPUs) and uses dynamic methods to shift computational work to optimize performance.

“This change lets us develop methods to optimize the code over time and not force the code to run on a specific implementation. In terms of optimization, we are enabling different levels of parallelization within and across tensor contraction expressions. This allows us to tune the parallelization strategy to the specific execution,” states Sriram Krishnamoorthy, a laboratory fellow and computer scientist at Pacific Northwest National Laboratory. The change will also enable NWChemEx to run on the Intel Ponte Vecchio GPU units that will be incorporated in Aurora.

The key points are modernization of the Density Function Theory (DFT) code that is widely used in quantum chemistry, and the Coupled Cluster (CC) theory that is used to access the high accuracy required for predictive chemistry. The team is focused on implementing reduced scaling chemistry algorithms that challenge implementation on high performance, heterogeneous hardware due to nonuniform sparsity in these methods.

David William-Young, a research scientist at Lawrence Berkeley National Laboratory, states, “To aid in the implementation, the team uses modularization of the algorithmic components (grid/batch generation, load balancing, local work drivers, reduction, etc.), so they can be swapped and sandboxed individually to rapidly test new ideas and architecture-specific implementations without interfering with other components. The module approach opens the door for reuse of software written for one purpose in other parts of NWChemEx.”

Resolving computational issues in NWChemEx

“New exascale hardware requires a new software environment. The team is modifying the code to keep up with all the changes in hardware architecture by adapting to a new and unifying programming model – oneAPI. The knowledge gained from the NWChemEx project can be used and shared when Aurora is operational,” states Abhishek Bagusetty, Aurora Early Science Program Postdoctoral Appointee, Argonne Leadership Computing Facility (ALCF).

ALCF is a U.S. Department of Energy (DOE) Office of Science User Facility located at Argonne National Laboratory. The ALCF team works with researchers to help enable breakthroughs in science and engineering by providing supercomputing resources and expertise to the research community.

“The team is working to resolve issues such as load-balancing occurring on the Oak Ridge National Laboratory Summit supercomputer in preparation for Aurora. Methods used to resolve communication and load balancing will apply outside of the NWChemEx code,” indicates Ajay Panyala, computer scientist at Pacific Northwest National Laboratory.

Using NWChemEx to find catalytic materials for conversion of biomass into biofuels

The development of advanced biofuels is driven by both energy security and climate change considerations. The Department of Energy (DOE) has an advanced biofuels program to develop fuels that can use the existing infrastructure and replace existing fuels with biofuels on a gallon-for-gallon basis. Windus states, “Producing high-quality biofuels in a sustainable and economically competitive way is technically challenging, especially in a changing global climate. The NWChemEx project directly addresses one of DOE’s priority goals to develop high-performance computational models demonstrating that biomass can be a viable, sustainable feedstock for the production of biofuels and other bioproducts. Of particular interest in this project is the prediction of specific, selective and low-temperature catalytic conversion of biomass to fuels and other products.”

Zeolites, such as H-ZSM-5, offer great promise for the catalytic conversion of renewable biomass-derived alcohols into fuels and chemicals. Compared to metal oxides with diverse surface and acid properties, acidic zeolites are solid structures that have relatively well-defined and uniform acid site structures. This makes it easy to perform rigorous kinetic and theoretical investigations of the effect of acid strength and solvation environment and confinement on the chemical reaction energies. Although there have been a number of prior atomic-scale computational studies of these systems, unraveling the true complexity of the conversion process and identifying means of achieving conversions at lower temperatures and pressures is an unsolved problem.

To illustrate the capability of NWChemEx for chemical reactions, the project will examine a number of elementary chemical transformations that have been postulated for the conversion of propanol to propene in the H-ZSM-5 zeolite (basic unit cell: Si96O192). The team will run calculations using computer simulations to predict the binding energy of water and propanol and their reactions in the zeolite cavity to help identify appropriate biofuels.

“By running the millions of simulations on Aurora, and then identifying a smaller subset of possible chemical solutions, the task of conducting physical experiments will be facilitated and therefore speed the discovery of real solutions,” states Windus.

Figure 1. Example of H-ZSM5 with 2-propanol and water. Image curtesy of Peter Sushko (PNNL).

Intel solutions used in developing NWChemEx

NWChemEx uses Data Parallel C++ (DPC++) as one of its programming models. DPC++ is a cross-architecture programming language, based on C++ and SYCL and part of oneAPI – an industry initiative to unify and simplify application development across diverse computing architectures. To facilitate porting of any existing CUDA code, Intel’s DPC++ Compatibility Tool helps migrate that code to create new DPC++ code. Further performance analysis and tuning uses Intel VTune Profiler and Intel Advisor. The compilers (C, C++, Fortran), Intel oneAPI Math Kernel Library (oneMKL) and Intel oneAPI DPC++ Library (oneDPL) – all part of Intel’s oneAPI Base Toolkit, are also employed.

Future benefits of NWChemEx and exascale supercomputers

The impact of the team’s biofuel research and rewriting NWChemEx software will have benefits outside the Argonne ESP project. “Chemistry simulations need to be able to model more complex, realistic systems. With current hardware and software, we can only simulate simplistic systems by removing impurities, getting rid of defects, and simplifying the solvent environments. In the future, researchers must be able to do real-life simulations that bring in all the complexities of the chemical system to have confidence the research results show the real-life issues, and the research findings will have comparable results to experimental conditions,” states Windus.

“Quantum chemistry research methods have become ubiquitous in hard science and engineering and are used in a variety of fields. What makes molecular dynamics and quantum chemistry tools such as NWChemEx so valuable is that they provide predictive results that are accurate when compared with many other research algorithms. Tools such as NWChemEx and access to hardware such as the future Argonne Aurora supercomputer will be critical to finding future solutions,” states Eric Bylaska, chemist at Pacific Northwest National Laboratory.

NWChemEx research was supported by the Exascale Computing Project, a collaborative effort of the U.S. Department of Energy Office of Science and the National Nuclear Security Administration.

Linda Barney is the founder and owner of Barney and Associates, a technical/marketing writing, training, and web design firm in Beaverton, Oregon.

Subscribe to HPCwire's Weekly Update!

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

AI Saves the Planet this Earth Day

April 22, 2024

Earth Day was originally conceived as a day of reflection. Our planet’s life-sustaining properties are unlike any other celestial body that we’ve observed, and this day of contemplation is meant to provide all of us Read more…

Intel Announces Hala Point – World’s Largest Neuromorphic System for Sustainable AI

April 22, 2024

As we find ourselves on the brink of a technological revolution, the need for efficient and sustainable computing solutions has never been more critical.  A computer system that can mimic the way humans process and s Read more…

Empowering High-Performance Computing for Artificial Intelligence

April 19, 2024

Artificial intelligence (AI) presents some of the most challenging demands in information technology, especially concerning computing power and data movement. As a result of these challenges, high-performance computing Read more…

Kathy Yelick on Post-Exascale Challenges

April 18, 2024

With the exascale era underway, the HPC community is already turning its attention to zettascale computing, the next of the 1,000-fold performance leaps that have occurred about once a decade. With this in mind, the ISC Read more…

2024 Winter Classic: Texas Two Step

April 18, 2024

Texas Tech University. Their middle name is ‘tech’, so it’s no surprise that they’ve been fielding not one, but two teams in the last three Winter Classic cluster competitions. Their teams, dubbed Matador and Red Read more…

2024 Winter Classic: The Return of Team Fayetteville

April 18, 2024

Hailing from Fayetteville, NC, Fayetteville State University stayed under the radar in their first Winter Classic competition in 2022. Solid students for sure, but not a lot of HPC experience. All good. They didn’t Read more…

AI Saves the Planet this Earth Day

April 22, 2024

Earth Day was originally conceived as a day of reflection. Our planet’s life-sustaining properties are unlike any other celestial body that we’ve observed, Read more…

Kathy Yelick on Post-Exascale Challenges

April 18, 2024

With the exascale era underway, the HPC community is already turning its attention to zettascale computing, the next of the 1,000-fold performance leaps that ha Read more…

Software Specialist Horizon Quantum to Build First-of-a-Kind Hardware Testbed

April 18, 2024

Horizon Quantum Computing, a Singapore-based quantum software start-up, announced today it would build its own testbed of quantum computers, starting with use o Read more…

MLCommons Launches New AI Safety Benchmark Initiative

April 16, 2024

MLCommons, organizer of the popular MLPerf benchmarking exercises (training and inference), is starting a new effort to benchmark AI Safety, one of the most pre Read more…

Exciting Updates From Stanford HAI’s Seventh Annual AI Index Report

April 15, 2024

As the AI revolution marches on, it is vital to continually reassess how this technology is reshaping our world. To that end, researchers at Stanford’s Instit Read more…

Intel’s Vision Advantage: Chips Are Available Off-the-Shelf

April 11, 2024

The chip market is facing a crisis: chip development is now concentrated in the hands of the few. A confluence of events this week reminded us how few chips Read more…

The VC View: Quantonation’s Deep Dive into Funding Quantum Start-ups

April 11, 2024

Yesterday Quantonation — which promotes itself as a one-of-a-kind venture capital (VC) company specializing in quantum science and deep physics  — announce Read more…

Nvidia’s GTC Is the New Intel IDF

April 9, 2024

After many years, Nvidia's GPU Technology Conference (GTC) was back in person and has become the conference for those who care about semiconductors and AI. I Read more…

Nvidia H100: Are 550,000 GPUs Enough for This Year?

August 17, 2023

The GPU Squeeze continues to place a premium on Nvidia H100 GPUs. In a recent Financial Times article, Nvidia reports that it expects to ship 550,000 of its lat Read more…

Synopsys Eats Ansys: Does HPC Get Indigestion?

February 8, 2024

Recently, it was announced that Synopsys is buying HPC tool developer Ansys. Started in Pittsburgh, Pa., in 1970 as Swanson Analysis Systems, Inc. (SASI) by John Swanson (and eventually renamed), Ansys serves the CAE (Computer Aided Engineering)/multiphysics engineering simulation market. Read more…

Intel’s Server and PC Chip Development Will Blur After 2025

January 15, 2024

Intel's dealing with much more than chip rivals breathing down its neck; it is simultaneously integrating a bevy of new technologies such as chiplets, artificia Read more…

Choosing the Right GPU for LLM Inference and Training

December 11, 2023

Accelerating the training and inference processes of deep learning models is crucial for unleashing their true potential and NVIDIA GPUs have emerged as a game- Read more…

Baidu Exits Quantum, Closely Following Alibaba’s Earlier Move

January 5, 2024

Reuters reported this week that Baidu, China’s giant e-commerce and services provider, is exiting the quantum computing development arena. Reuters reported � Read more…

Comparing NVIDIA A100 and NVIDIA L40S: Which GPU is Ideal for AI and Graphics-Intensive Workloads?

October 30, 2023

With long lead times for the NVIDIA H100 and A100 GPUs, many organizations are looking at the new NVIDIA L40S GPU, which it’s a new GPU optimized for AI and g Read more…

Shutterstock 1179408610

Google Addresses the Mysteries of Its Hypercomputer 

December 28, 2023

When Google launched its Hypercomputer earlier this month (December 2023), the first reaction was, "Say what?" It turns out that the Hypercomputer is Google's t Read more…

AMD MI3000A

How AMD May Get Across the CUDA Moat

October 5, 2023

When discussing GenAI, the term "GPU" almost always enters the conversation and the topic often moves toward performance and access. Interestingly, the word "GPU" is assumed to mean "Nvidia" products. (As an aside, the popular Nvidia hardware used in GenAI are not technically... Read more…

Leading Solution Providers

Contributors

Shutterstock 1606064203

Meta’s Zuckerberg Puts Its AI Future in the Hands of 600,000 GPUs

January 25, 2024

In under two minutes, Meta's CEO, Mark Zuckerberg, laid out the company's AI plans, which included a plan to build an artificial intelligence system with the eq Read more…

China Is All In on a RISC-V Future

January 8, 2024

The state of RISC-V in China was discussed in a recent report released by the Jamestown Foundation, a Washington, D.C.-based think tank. The report, entitled "E Read more…

Shutterstock 1285747942

AMD’s Horsepower-packed MI300X GPU Beats Nvidia’s Upcoming H200

December 7, 2023

AMD and Nvidia are locked in an AI performance battle – much like the gaming GPU performance clash the companies have waged for decades. AMD has claimed it Read more…

Nvidia’s New Blackwell GPU Can Train AI Models with Trillions of Parameters

March 18, 2024

Nvidia's latest and fastest GPU, codenamed Blackwell, is here and will underpin the company's AI plans this year. The chip offers performance improvements from Read more…

Eyes on the Quantum Prize – D-Wave Says its Time is Now

January 30, 2024

Early quantum computing pioneer D-Wave again asserted – that at least for D-Wave – the commercial quantum era has begun. Speaking at its first in-person Ana Read more…

GenAI Having Major Impact on Data Culture, Survey Says

February 21, 2024

While 2023 was the year of GenAI, the adoption rates for GenAI did not match expectations. Most organizations are continuing to invest in GenAI but are yet to Read more…

The GenAI Datacenter Squeeze Is Here

February 1, 2024

The immediate effect of the GenAI GPU Squeeze was to reduce availability, either direct purchase or cloud access, increase cost, and push demand through the roof. A secondary issue has been developing over the last several years. Even though your organization secured several racks... Read more…

Intel’s Xeon General Manager Talks about Server Chips 

January 2, 2024

Intel is talking data-center growth and is done digging graves for its dead enterprise products, including GPUs, storage, and networking products, which fell to Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire