A new open source multi-GPU 2D flood model called TRITON is presented in this work. The model solves the 2D shallow water equations with source terms using a time-explicit first order upwind scheme based on an Augmented Roe's solver that incorporates a careful estimation of bed strengths and a local implicit formulation of friction terms. The scheme is demonstrated to be first order accurate, robust and able to solve for flows under various conditions. TRITON is implemented such that the model effectively utilizes heterogeneous architectures, from single to multiple CPUs and GPUs. Different test cases are shown to illustrate the capabilities and performance of the model, showing promising runtimes for large spatial and temporal scales when leveraging the computer power of GPUs. Under this hardware configuration, communication and input/output subroutines may impact the scalability. The code is developed under an open source license and can be freely downloaded in https://code.ornl.gov/hydro/triton.

这项工作提出了一种新的开源多GPU 2D洪水模型TRITON。该模型使用基于增强Roe求解器的时间显式一阶迎风方案使用源项来求解二维浅水方程，该方案结合了对床层强度的仔细估计和摩擦项的局部隐式公式。该方案被证明是一阶准确，稳健的，并且能够解决各种条件下的流量。实现TRITON的目的是使该模型有效地利用从单个CPU到多个CPU和GPU的异构体系结构。展示了不同的测试用例以说明该模型的功能和性能，并展示了利用GPU的计算机功能在大的空间和时间范围上具有希望的运行时。在这种硬件配置下，通信和输入/输出子例程可能会影响可伸缩性。该代码是在开放源代码许可下开发的，可以在https://code.ornl.gov/hydro/triton中免费下载。