High-performance task-based water balance modeling

Water balance modeling plays a pivotal role in sustainable water management, as it underpins the understanding of hydrological processes that govern resource distribution, ecosystem stability, and long-term environmental planning. Accurate and efficient computational tools are essential to capture the spatial and temporal dynamics of water balance, particularly in complex geological and urban environments. WaterpyBal is an innovative modeling framework specifically designed to construct spatial-temporal water balance models. It effectively integrates multiple stages of hydrological assessment-including data interpolation, evapotranspiration estimation, and infiltration computation-while accounting for soil heterogeneity and components of the urban water cycle. The tool demonstrates robust performance when applied to both synthetic and experimental datasets, providing reliable and scalable results.

In the context of the exascale era, where data-intensive environmental models demand unprecedented computational power, High-Performance Computing (HPC) frameworks are essential to ensure scalability and efficiency. To this end, WaterpyBal has been enhanced through its integration with PyCOMPSs, the Python binding of the COMPSs programming model. PyCOMPSs enables the transparent parallelization of Python applications by identifying task-level parallelism through annotated methods and dynamically constructing a task-dependency graph during runtime. This graph-driven execution model allows efficient scheduling and data management across distributed computing infrastructures such as clusters and cloud platforms.

The integration of WaterpyBal with PyCOMPSs significantly improves its computational performance, enabling the simulation of large-scale, high-resolution water balance models within feasible timeframes. This work demonstrates the potential of combining advanced hydrological modeling with state-of-the-art parallel computing frameworks to address emerging challenges in environmental modeling and resource management at scale.