GPU 加快颱風模擬：4 張 A100 對 128 粒 CPU，速度數字點解有兩個版本
xNUMA 測試交出效能同耗電數據，準確度仍要另外驗證
GPU 今次跑物理模型，唔關生成式 AI 事
TechNews 科技新報報道，美國海軍研究院團隊將 xNUMA 非靜力大氣模型搬上 GPU，再用 NCSA 嘅 Delta 超級電腦模擬熱帶氣旋快速增強。GPU 呢幾年幾乎同 AI 綁埋一齊講，不過今次做嘅係另一類工作：將大氣切成大量網格，同步計算流體、溫度同壓力點樣變。呢種大量重複又可以平行處理嘅數學，本身就啱 GPU 發揮。
圖片：Kang 等人／NCSA
「快 5 倍」同論文數字對唔上
NCSA 公布話，4 張 NVIDIA A100 對 128 粒 AMD EPYC 7763 CPU，模擬時間約快 5 倍，能源效率亦高約 5 倍。不過正式論文交代得仔細啲：同一項效能測試入面，CPU 每個 time step 要 5.563 秒，GPU 只要 0.536 秒，實際係約 10.4 倍速度差距；每步能源用量就係 3,115.3 焦耳對 643.2 焦耳，相差約 4.84 倍。兩邊冇解釋點解速度數字有落差，所以「5 倍」只適合當 NCSA 對外文章嘅概括講法。
個 benchmark 有清楚邊界
論文用咗一個約 1.66 億自由度嘅網格，量度初始化後首 100 個 time steps，亦即只計主要時間積分部分。CPU 版本本身已針對 CPU 調校，GPU 版本就用 OpenACC 將核心運算搬去 A100。研究團隊估計，GPU 浮點運算能力較高，加上只用 4 張卡，處理器之間交換資料嘅負擔低過 128 粒 CPU。呢個結果只代表 xNUMA、呢組硬件同呢個測試設定，換模型、GPU 型號、網格大小或者通訊架構，差距都可能完全唔同。
仲有一點要留神：論文嘅能源數字用硬件規格標示嘅最高功耗推算，唔係用電錶量度成套系統。佢可以反映完成同一段運算所需嘅大概能源差距，但未計冷卻、儲存、網絡設備同其他機房開支。研究證明 GPU 版本喺指定條件下慳時間兼慳電，但唔可以由呢次測試推論所有氣象運算中心嘅總成本。
模擬到快速增強，未等於識得預報
團隊以 1997 年颶風 Guillermo 嘅機載多普勒雷達資料，將觀測所得嘅潛熱加熱分布輸入一個理想化氣旋，再模擬六小時演變。模型量到切向風速增加超過每秒 20 米，亦重現到快速增強嘅特徵。不過設定入面冇背景風，關鍵加熱資料亦預先輸入模型；呢個比較似用已知資料重建氣旋嘅物理變化，同每日靠未必齊全嘅觀測，即時計算未來路徑同強度嘅業務預報，仲有一段距離。
運算速度、能源用量同預報準確度要分開睇。 跑快咗可以畀氣象中心提高解像度、試更多初始條件，或者喺同一時限內跑更多 ensemble 成員，間接增加預報員手上嘅資訊。不過準唔準仍然受初始觀測、資料同化、雲雨物理、海氣互動同模型偏差影響。今次研究驗證咗 xNUMA 喺 GPU 上跑得有效率，冇直接證明佢預測未知氣旋時會準過現役模型。
對香港有潛在價值，但暫時未見採用
快速增強會壓縮預警同防風部署時間，計算快幾倍當然有實際吸引力。香港天文台公開資料顯示，2024 年投入服務嘅 HPC 系統運算能力達每秒 1.3 千萬億次浮點運算，可支援較高解像度數值預報，同時用唔同設定多跑幾次，幫手處理熱帶氣旋同暴雨嘅概率預報。不過公開資料暫時冇提過天文台採用 xNUMA 或呢套 GPU 移植方案，現階段只可以話技術方向同本地預報需求相符。
呢項工作較貼切嘅定位係一個有份量嘅工程驗證：舊有大氣模型可以搬上 GPU，而且喺特定測試慳到明顯時間同能源。下一步要睇完整業務預報流程嘅測試，包括資料同化、較長預報時段、更多氣旋個案，同現役模型直接比較誤差，先可以判斷佢會唔會真正改善預警。
參考來源
- TechNews 科技新報 — GPU 助攻氣象預報大突破：熱帶氣旋模擬提速 5 倍、耗能降 5 倍 — original report
- Modeling Rapid Intensification of a Tropical Cyclone with GPUs — NCSA 官方文章，確認 Delta、4 張 A100 對 128 粒 CPU、Guillermo 個案同約 5 倍講法。
- A GPU-accelerated simulation of rapid intensification of a tropical cyclone with observed heating — 正式研究論文頁，核對模型方法、發表狀態同研究範圍。
- A GPU-accelerated simulation of rapid intensification of a tropical cyclone with observed heating（arXiv 全文） — 公開論文全文，核對 10.4 倍速度差距、4.84 倍能源差距、測試設定同能源推算方法。
- Multiscale Modeling Framework Using Element-Based Galerkin Methods for Moist Atmospheric Limited-Area Simulations — xNUMA 相關基礎研究，交代模型架構、多尺度用途同早期驗證範圍。
- The Hong Kong Observatory Computer System and Its Applications — 天文台官方資料，核對 2024 年 HPC 系統、運算能力同熱帶氣旋概率預報用途。
本文根據原文及公開資料整理；資料有出入時，以原文及官方資料為準。
