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 industy updates delivered to you every week!

With New Owner and New Roadmap, an Independent Omni-Path Is Staging a Comeback

July 23, 2021

Put on a shelf by Intel in 2019, Omni-Path faced a uncertain future, but under new custodian Cornelis Networks, OmniPath is looking to make a comeback as an independent high-performance interconnect solution. A "significant refresh" – called Omni-Path Express – is coming later this year according to the company. Cornelis Networks formed last September as a spinout of Intel's Omni-Path division. Read more…

PEARC21 Panel Reviews Eight New NSF-Funded HPC Systems Debuting in 2021

July 23, 2021

Over the past few years, the NSF has funded a number of HPC systems to further supply the open research community with computational resources to meet that community’s changing and expanding needs. A review of these systems at the PEARC21 conference (July 19-22) highlighted... Read more…

Chameleon’s HPC Testbed Sharpens Its Edge, Presses ‘Replay’

July 22, 2021

“One way of saying what I do for a living is to say that I develop scientific instruments,” said Kate Keahey, a senior fellow at the University of Chicago and a computer scientist at Argonne National Laboratory, as s Read more…

PEARC21 Plenary Session: AI for Innovative Social Work

July 21, 2021

AI analysis of social media poses a double-edged sword for social work and addressing the needs of at-risk youths, said Desmond Upton Patton, senior associate dean, Innovation and Academic Affairs, Columbia University. S Read more…

Summer Reading: “High-Performance Computing Is at an Inflection Point”

July 21, 2021

At last month’s 11th International Symposium on Highly Efficient Accelerators and Reconfigurable Technologies (HEART), a group of researchers led by Martin Schulz of the Leibniz Supercomputing Center (Munich) presented a “position paper” in which they argue HPC architectural landscape... Read more…

AWS Solution Channel

Accelerate innovation in healthcare and life sciences with AWS HPC

With Amazon Web Services, researchers can access purpose-built HPC tools and services along with scientific and technical expertise to accelerate the pace of discovery. Whether you are sequencing the human genome, using AI/ML for disease detection or running molecular dynamics simulations to develop lifesaving drugs, AWS has the infrastructure you need to run your HPC workloads. Read more…

PEARC21 Panel: Wafer-Scale-Engine Technology Accelerates Machine Learning, HPC

July 21, 2021

Early use of Cerebras’ CS-1 server and wafer-scale engine (WSE) has demonstrated promising acceleration of machine-learning algorithms, according to participants in the Scientific Research Enabled by CS-1 Systems panel Read more…

With New Owner and New Roadmap, an Independent Omni-Path Is Staging a Comeback

July 23, 2021

Put on a shelf by Intel in 2019, Omni-Path faced a uncertain future, but under new custodian Cornelis Networks, OmniPath is looking to make a comeback as an independent high-performance interconnect solution. A "significant refresh" – called Omni-Path Express – is coming later this year according to the company. Cornelis Networks formed last September as a spinout of Intel's Omni-Path division. Read more…

Chameleon’s HPC Testbed Sharpens Its Edge, Presses ‘Replay’

July 22, 2021

“One way of saying what I do for a living is to say that I develop scientific instruments,” said Kate Keahey, a senior fellow at the University of Chicago a Read more…

Summer Reading: “High-Performance Computing Is at an Inflection Point”

July 21, 2021

At last month’s 11th International Symposium on Highly Efficient Accelerators and Reconfigurable Technologies (HEART), a group of researchers led by Martin Schulz of the Leibniz Supercomputing Center (Munich) presented a “position paper” in which they argue HPC architectural landscape... Read more…

PEARC21 Panel: Wafer-Scale-Engine Technology Accelerates Machine Learning, HPC

July 21, 2021

Early use of Cerebras’ CS-1 server and wafer-scale engine (WSE) has demonstrated promising acceleration of machine-learning algorithms, according to participa Read more…

15 Years Later, the Green500 Continues Its Push for Energy Efficiency as a First-Order Concern in HPC

July 15, 2021

The Green500 list, which ranks the most energy-efficient supercomputers in the world, has virtually always faced an uphill battle. As Wu Feng – custodian of the Green500 list and an associate professor at Virginia Tech – tells it, “noone" cared about energy efficiency in the early 2000s, when the seeds... Read more…

Frontier to Meet 20MW Exascale Power Target Set by DARPA in 2008

July 14, 2021

After more than a decade of planning, the United States’ first exascale computer, Frontier, is set to arrive at Oak Ridge National Laboratory (ORNL) later this year. Crossing this “1,000x” horizon required overcoming four major challenges: power demand, reliability, extreme parallelism and data movement. Read more…

Quantum Roundup: IBM, Rigetti, Phasecraft, Oxford QC, China, and More

July 13, 2021

IBM yesterday announced a proof for a quantum ML algorithm. A week ago, it unveiled a new topology for its quantum processors. Last Friday, the Technical Univer Read more…

ExaWind Prepares for New Architectures, Bigger Simulations

July 10, 2021

The ExaWind project describes itself in terms of terms like wake formation, turbine-turbine interaction and blade-boundary-layer dynamics, but the pitch to the Read more…

AMD Chipmaker TSMC to Use AMD Chips for Chipmaking

May 8, 2021

TSMC has tapped AMD to support its major manufacturing and R&D workloads. AMD will provide its Epyc Rome 7702P CPUs – with 64 cores operating at a base cl Read more…

Intel Launches 10nm ‘Ice Lake’ Datacenter CPU with Up to 40 Cores

April 6, 2021

The wait is over. Today Intel officially launched its 10nm datacenter CPU, the third-generation Intel Xeon Scalable processor, codenamed Ice Lake. With up to 40 Read more…

Berkeley Lab Debuts Perlmutter, World’s Fastest AI Supercomputer

May 27, 2021

A ribbon-cutting ceremony held virtually at Berkeley Lab's National Energy Research Scientific Computing Center (NERSC) today marked the official launch of Perlmutter – aka NERSC-9 – the GPU-accelerated supercomputer built by HPE in partnership with Nvidia and AMD. Read more…

Ahead of ‘Dojo,’ Tesla Reveals Its Massive Precursor Supercomputer

June 22, 2021

In spring 2019, Tesla made cryptic reference to a project called Dojo, a “super-powerful training computer” for video data processing. Then, in summer 2020, Tesla CEO Elon Musk tweeted: “Tesla is developing a [neural network] training computer called Dojo to process truly vast amounts of video data. It’s a beast! … A truly useful exaflop at de facto FP32.” Read more…

Google Launches TPU v4 AI Chips

May 20, 2021

Google CEO Sundar Pichai spoke for only one minute and 42 seconds about the company’s latest TPU v4 Tensor Processing Units during his keynote at the Google I Read more…

CentOS Replacement Rocky Linux Is Now in GA and Under Independent Control

June 21, 2021

The Rocky Enterprise Software Foundation (RESF) is announcing the general availability of Rocky Linux, release 8.4, designed as a drop-in replacement for the soon-to-be discontinued CentOS. The GA release is launching six-and-a-half months after Red Hat deprecated its support for the widely popular, free CentOS server operating system. The Rocky Linux development effort... Read more…

CERN Is Betting Big on Exascale

April 1, 2021

The European Organization for Nuclear Research (CERN) involves 23 countries, 15,000 researchers, billions of dollars a year, and the biggest machine in the worl Read more…

Iran Gains HPC Capabilities with Launch of ‘Simorgh’ Supercomputer

May 18, 2021

Iran is said to be developing domestic supercomputing technology to advance the processing of scientific, economic, political and military data, and to strengthen the nation’s position in the age of AI and big data. On Sunday, Iran unveiled the Simorgh supercomputer, which will deliver.... Read more…

Leading Solution Providers

Contributors

HPE Launches Storage Line Loaded with IBM’s Spectrum Scale File System

April 6, 2021

HPE today launched a new family of storage solutions bundled with IBM’s Spectrum Scale Erasure Code Edition parallel file system (description below) and featu Read more…

Julia Update: Adoption Keeps Climbing; Is It a Python Challenger?

January 13, 2021

The rapid adoption of Julia, the open source, high level programing language with roots at MIT, shows no sign of slowing according to data from Julialang.org. I Read more…

10nm, 7nm, 5nm…. Should the Chip Nanometer Metric Be Replaced?

June 1, 2020

The biggest cool factor in server chips is the nanometer. AMD beating Intel to a CPU built on a 7nm process node* – with 5nm and 3nm on the way – has been i Read more…

GTC21: Nvidia Launches cuQuantum; Dips a Toe in Quantum Computing

April 13, 2021

Yesterday Nvidia officially dipped a toe into quantum computing with the launch of cuQuantum SDK, a development platform for simulating quantum circuits on GPU-accelerated systems. As Nvidia CEO Jensen Huang emphasized in his keynote, Nvidia doesn’t plan to build... Read more…

Microsoft to Provide World’s Most Powerful Weather & Climate Supercomputer for UK’s Met Office

April 22, 2021

More than 14 months ago, the UK government announced plans to invest £1.2 billion ($1.56 billion) into weather and climate supercomputing, including procuremen Read more…

Q&A with Jim Keller, CTO of Tenstorrent, and an HPCwire Person to Watch in 2021

April 22, 2021

As part of our HPCwire Person to Watch series, we are happy to present our interview with Jim Keller, president and chief technology officer of Tenstorrent. One of the top chip architects of our time, Keller has had an impactful career. Read more…

Quantum Roundup: IBM, Rigetti, Phasecraft, Oxford QC, China, and More

July 13, 2021

IBM yesterday announced a proof for a quantum ML algorithm. A week ago, it unveiled a new topology for its quantum processors. Last Friday, the Technical Univer Read more…

Senate Debate on Bill to Remake NSF – the Endless Frontier Act – Begins

May 18, 2021

The U.S. Senate today opened floor debate on the Endless Frontier Act which seeks to remake and expand the National Science Foundation by creating a technology Read more…

  • arrow
  • Click Here for More Headlines
  • arrow
HPCwire