Weather at Exascale: Load Balancing for Heterogeneous Systems

By Oliver Peckham

March 30, 2020

The first months of 2020 were dominated by weather and climate supercomputing news, with major announcements coming from the UK, the European Centre for Medium-Range Weather Forecasts and the U.S. National Oceanic and Atmospheric Administration. Richard Loft, director of technology development at the Computational Information Systems Laboratory of the NSF-supported National Center for Atmospheric Research (NCAR), took the (virtual) stage at Nvidia’s GTC 2020 to highlight NCAR’s progress in moving weather and climate supercomputing into the exascale era.

Specifically, Loft discussed NCAR’s Model for Prediction Across Scales (MPAS), its latest-generation weather model. MPAS, developed in partnership with Los Alamos National Laboratory, is the basis for the Global High-Resolution Forecasting System (GRAF) introduced last year by IBM and its subsidiary The Weather Company and running in production since October. Loft, who noted that the team was funding-limited and relied heavily on student support, called GRAF the “crowning achievement” of MPAS so far. 

“This partnership between NCAR, IBM and the Weather Company,” Loft said, “has put into production a model with three-kilometer resolution over lots of areas in the world – and especially the developing world – where they haven’t had those kinds of resolutions before.” These new capabilities, he said, would enable significantly enhanced weather forecast products for those underserved areas.

And, crucially: “[GRAF] is the first GPU-based global forecast model, to my knowledge, in production anywhere in the world at this point.”

The goals for MPAS

In his talk, Loft outlined the objectives NCAR is aiming for with MPAS as the computing world approaches the exascale era – and how load balancing between GPUs and CPUs factors into those objectives, with GPUs becoming an increasingly important part of many major supercomputers. “So when we started this over three years ago,” Loft said, “we started looking at it from the perspective of trying to achieve a number of core capabilities for this model.”

First was performance portability. “We wanted to get the best performance on GPUs we could, but we wanted to maintain performance on CPUs,” Loft said. “We didn’t want to sacrifice that, because that’s our bread and butter.” 

Second was resilience. “We needed to have a strategy for load-balancing all the code on the hybrid node,” he explained, “in such a way that it wasn’t too hard-wired for any one combination of CPU and GPU.”

Finally, the researchers wanted to port the minimum amount of code. For this, Loft said, they “triaged” the process. They ported most of the physics and dynamics code to the GPUs (using OpenACC), but left the radiation code (which accounts for the transfer of solar radiation through the atmosphere) as a CPU-only code. They made this choice, Loft explained, because the radiative code was over 30,000 lines. Radiation, Loft said, was “expensive”: “If you called it every timestep, the model would grind to a stop.” Fortunately, radiation also evolved on a much slower timescale, so operating it asynchronously from the rest of the model was sensible on a sheer physical level.

The researchers also left the included land surface model – which accounts for water and energy transfers between land and the atmosphere – as a CPU-only model, since it was over 20,000 lines of “branchy code” and didn’t take much time to run (unfortunately, though, Loft said, it had to be run synchronously).

Tinkering to perfection on two powerful supercomputers

The Summit supercomputer.

To optimize its newly ported model, NCAR turned to a pair of supercomputers. First was NCAR’s own Cheyenne system, which houses 4,032 nodes (each with Intel Xeon Broadwell processors), an aggregate 313 TB of memory, connected by Mellanox EDR InfiniBand. Second was Summit at Oak Ridge National Laboratory, housing 4,608 nodes (each with two IBM Power9 CPUs and six Nvidia Volta GPUs), over 10 PB of memory, tied together with EDR InfiniBand. 

At 4.8 and 148.6 Linpack petaflops, Cheyenne and Summit respectively placed 44th and 1st on the most recent Top500 list of the world’s most powerful publicly ranked supercomputers. The two systems were carefully selected: Cheyenne was selected for its CPU-only homogeneous design, while Summit offered a glimpse into the future of heterogeneous, GPU-heavy system design. Summit also bore similarity to GRAF’s IBM Power9-based supercomputer, which features 84 nodes with four Nvidia V100 GPUs each.

The supercomputers were employed to help NCAR find the “Goldilocks zone” for resource allocation. When allocating CPU cores and/or GPUs to either the radiation or the dynamics or the radiation, a mismatch in the timing could lead to serious losses of time and money. Allocate too little to radiation on the CPUs, for instance, and the GPUs powering integration (the remainder of the model) would have to pause their work while waiting for the CPUs to catch up.

An illustration of the possible mismatches between model timing. Image courtesy of NCAR.

For the tests, the researchers leveraged 76 CPU nodes on Cheyenne and 76 hybrid nodes on Summit to run MPAS at a ten-kilometer resolution, throwing 81,920 points of analysis to each node on each system. They tested balances in CPU and GPU allocation, changes in symmetric multithreading and changes in the interval at which radiation was re-integrated into the model. 

On Cheyenne, the researchers found that the system was integration-limited, performing relatively poorly on general model tasks – so, they assigned most of the cores on each CPU to the general model and very few to radiation (a 2:1 ratio). On Summit, meanwhile, they found that with the GPUs contributing massive amounts to the processing power, the system was radiation-limited. Eventually, the researchers found the best performance at a balance of 18 CPU cores (and the GPUs) dedicated to the integrated model, with 24 dedicated to radiation.

Summit’s configuration for MPAS. Image courtesy of NCAR.

In addition to finding these settings for optimization, the researchers saw three times the throughput for MPAS when using GPU nodes compared to the CPU nodes. In terms of the frequency of re-integrating the radiation, Loft said that on Summit, they found that “there is insufficient CPU power to keep up with six GPUs unless the radiation [was] called pretty infrequently.”

The path forward

Next, the NCAR team will work on further optimizing message passing, as well as porting the land surface model to GPUs to free up even more cores for radiation processing. (Loft says that they’re considering porting the radiation code to GPUs, but only as a “last resort.) 

While Loft acknowledges that there’s work to be done on scalability and throughput, he says the model is well on its way to being optimized. “It’s a work in progress, but we’ve got the full model working,” Loft said. “It’s a good first cut.”

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!

Hats Over Hearts: Remembering Rich Brueckner

May 26, 2020

It is with great sadness that we announce the death of Rich Brueckner. His passing is an unexpected and enormous blow to both his family and our HPC family. Rich was born in Milwaukee, Wisconsin on April 12, 1962. His Read more…

Supercomputer Simulations Reveal the Fate of the Neanderthals

May 25, 2020

For hundreds of thousands of years, neanderthals roamed the planet, eventually (almost 50,000 years ago) giving way to homo sapiens, which quickly became the dominant primate species, with the neanderthals disappearing b Read more…

By Oliver Peckham

Discovering Alternative Solar Panel Materials with Supercomputing

May 23, 2020

Solar power is quickly growing in the world’s energy mix, but silicon – a crucial material in the construction of photovoltaic solar panels – remains expensive, hindering solar’s expansion and competitiveness wit Read more…

By Oliver Peckham

Nvidia Q1 Earnings Top Expectations, Datacenter Revenue Breaks $1B

May 22, 2020

Nvidia’s seemingly endless roll continued in the first quarter with the company announcing blockbuster earnings that exceeded Wall Street expectations. Nvidia said revenues for the period ended April 26 were up 39 perc Read more…

By Doug Black

TACC Supercomputers Delve into COVID-19’s Spike Protein

May 22, 2020

If you’ve been following COVID-19 research, by now, you’ve probably heard of the spike protein (or S-protein). The spike protein – which gives COVID-19 its namesake crown-like shape – is the virus’ crowbar into Read more…

By Oliver Peckham

AWS Solution Channel

Computational Fluid Dynamics on AWS

Over the past 30 years Computational Fluid Dynamics (CFD) has grown to become a key part of many engineering design processes. From aircraft design to modelling the blood flow in our bodies, the ability to understand the behaviour of fluids has enabled countless innovations and improved the time to market for many products. Read more…

Using HPC, Researchers Discover How Easily Hurricanes Form

May 21, 2020

Hurricane formation has long remained shrouded in mystery, with meteorologists unable to discern exactly what forces cause the devastating storms (also known as tropical cyclones) to materialize. Now, researchers at Flor Read more…

By Oliver Peckham

Microsoft’s Massive AI Supercomputer on Azure: 285k CPU Cores, 10k GPUs

May 20, 2020

Microsoft has unveiled a supercomputing monster – among the world’s five most powerful, according to the company – aimed at what is known in scientific an Read more…

By Doug Black

HPC in Life Sciences 2020 Part 1: Rise of AMD, Data Management’s Wild West, More 

May 20, 2020

Given the disruption caused by the COVID-19 pandemic and the massive enlistment of major HPC resources to fight the pandemic, it is especially appropriate to re Read more…

By John Russell

AMD Epyc Rome Picked for New Nvidia DGX, but HGX Preserves Intel Option

May 19, 2020

AMD continues to make inroads into the datacenter with its second-generation Epyc "Rome" processor, which last week scored a win with Nvidia's announcement that Read more…

By Tiffany Trader

Hacking Streak Forces European Supercomputers Offline in Midst of COVID-19 Research Effort

May 18, 2020

This week, a number of European supercomputers discovered intrusive malware hosted on their systems. Now, in the midst of a massive supercomputing research effo Read more…

By Oliver Peckham

Nvidia’s Ampere A100 GPU: Up to 2.5X the HPC, 20X the AI

May 14, 2020

Nvidia's first Ampere-based graphics card, the A100 GPU, packs a whopping 54 billion transistors on 826mm2 of silicon, making it the world's largest seven-nanom Read more…

By Tiffany Trader

Wafer-Scale Engine AI Supercomputer Is Fighting COVID-19

May 13, 2020

Seemingly every supercomputer in the world is allied in the fight against the coronavirus pandemic – but not many of them are fresh out of the box. Cerebras S Read more…

By Oliver Peckham

Startup MemVerge on Memory-centric Mission

May 12, 2020

Memory situated at the center of the computing universe, replacing processing, has long been envisioned as instrumental to radically improved datacenter systems Read more…

By Doug Black

In Australia, HPC Illuminates the Early Universe

May 11, 2020

Many billions of years ago, the universe was a swirling pool of gas. Unraveling the story of how we got from there to here isn’t an easy task, with many simul Read more…

By Oliver Peckham

Supercomputer Modeling Tests How COVID-19 Spreads in Grocery Stores

April 8, 2020

In the COVID-19 era, many people are treating simple activities like getting gas or groceries with caution as they try to heed social distancing mandates and protect their own health. Still, significant uncertainty surrounds the relative risk of different activities, and conflicting information is prevalent. A team of Finnish researchers set out to address some of these uncertainties by... Read more…

By Oliver Peckham

[email protected] Turns Its Massive Crowdsourced Computer Network Against COVID-19

March 16, 2020

For gamers, fighting against a global crisis is usually pure fantasy – but now, it’s looking more like a reality. As supercomputers around the world spin up Read more…

By Oliver Peckham

[email protected] Rallies a Legion of Computers Against the Coronavirus

March 24, 2020

Last week, we highlighted [email protected], a massive, crowdsourced computer network that has turned its resources against the coronavirus pandemic sweeping the globe – but [email protected] isn’t the only game in town. The internet is buzzing with crowdsourced computing... Read more…

By Oliver Peckham

Global Supercomputing Is Mobilizing Against COVID-19

March 12, 2020

Tech has been taking some heavy losses from the coronavirus pandemic. Global supply chains have been disrupted, virtually every major tech conference taking place over the next few months has been canceled... Read more…

By Oliver Peckham

DoE Expands on Role of COVID-19 Supercomputing Consortium

March 25, 2020

After announcing the launch of the COVID-19 High Performance Computing Consortium on Sunday, the Department of Energy yesterday provided more details on its sco Read more…

By John Russell

Steve Scott Lays Out HPE-Cray Blended Product Roadmap

March 11, 2020

Last week, the day before the El Capitan processor disclosures were made at HPE's new headquarters in San Jose, Steve Scott (CTO for HPC & AI at HPE, and former Cray CTO) was on-hand at the Rice Oil & Gas HPC conference in Houston. He was there to discuss the HPE-Cray transition and blended roadmap, as well as his favorite topic, Cray's eighth-gen networking technology, Slingshot. Read more…

By Tiffany Trader

Honeywell’s Big Bet on Trapped Ion Quantum Computing

April 7, 2020

Honeywell doesn’t spring to mind when thinking of quantum computing pioneers, but a decade ago the high-tech conglomerate better known for its control systems waded deliberately into the then calmer quantum computing (QC) waters. Fast forward to March when Honeywell announced plans to introduce an ion trap-based quantum computer whose ‘performance’ would... Read more…

By John Russell

Fujitsu A64FX Supercomputer to Be Deployed at Nagoya University This Summer

February 3, 2020

Japanese tech giant Fujitsu announced today that it will supply Nagoya University Information Technology Center with the first commercial supercomputer powered Read more…

By Tiffany Trader

Leading Solution Providers

SC 2019 Virtual Booth Video Tour

AMD
AMD
ASROCK RACK
ASROCK RACK
AWS
AWS
CEJN
CJEN
CRAY
CRAY
DDN
DDN
DELL EMC
DELL EMC
IBM
IBM
MELLANOX
MELLANOX
ONE STOP SYSTEMS
ONE STOP SYSTEMS
PANASAS
PANASAS
SIX NINES IT
SIX NINES IT
VERNE GLOBAL
VERNE GLOBAL
WEKAIO
WEKAIO

Contributors

Tech Conferences Are Being Canceled Due to Coronavirus

March 3, 2020

Several conferences scheduled to take place in the coming weeks, including Nvidia’s GPU Technology Conference (GTC) and the Strata Data + AI conference, have Read more…

By Alex Woodie

Exascale Watch: El Capitan Will Use AMD CPUs & GPUs to Reach 2 Exaflops

March 4, 2020

HPE and its collaborators reported today that El Capitan, the forthcoming exascale supercomputer to be sited at Lawrence Livermore National Laboratory and serve Read more…

By John Russell

‘Billion Molecules Against COVID-19’ Challenge to Launch with Massive Supercomputing Support

April 22, 2020

Around the world, supercomputing centers have spun up and opened their doors for COVID-19 research in what may be the most unified supercomputing effort in hist Read more…

By Oliver Peckham

Cray to Provide NOAA with Two AMD-Powered Supercomputers

February 24, 2020

The United States’ National Oceanic and Atmospheric Administration (NOAA) last week announced plans for a major refresh of its operational weather forecasting supercomputers, part of a 10-year, $505.2 million program, which will secure two HPE-Cray systems for NOAA’s National Weather Service to be fielded later this year and put into production in early 2022. Read more…

By Tiffany Trader

Summit Supercomputer is Already Making its Mark on Science

September 20, 2018

Summit, now the fastest supercomputer in the world, is quickly making its mark in science – five of the six finalists just announced for the prestigious 2018 Read more…

By John Russell

15 Slides on Programming Aurora and Exascale Systems

May 7, 2020

Sometime in 2021, Aurora, the first planned U.S. exascale system, is scheduled to be fired up at Argonne National Laboratory. Cray (now HPE) and Intel are the k Read more…

By John Russell

TACC Supercomputers Run Simulations Illuminating COVID-19, DNA Replication

March 19, 2020

As supercomputers around the world spin up to combat the coronavirus, the Texas Advanced Computing Center (TACC) is announcing results that may help to illumina Read more…

By Staff report

Nvidia’s Ampere A100 GPU: Up to 2.5X the HPC, 20X the AI

May 14, 2020

Nvidia's first Ampere-based graphics card, the A100 GPU, packs a whopping 54 billion transistors on 826mm2 of silicon, making it the world's largest seven-nanom Read more…

By Tiffany Trader

  • arrow
  • Click Here for More Headlines
  • arrow
Do NOT follow this link or you will be banned from the site!
Share This