polaris.ModelStep.update_io_tasks_config

ModelStep.update_io_tasks_config(config_model=None)

Modify model config options so the number of IO tasks and the stride between them are consistent with how MPI tasks are distributed across nodes (one IO task per node).

The number of tasks per node is based on the machine’s resources from mache (cores and GPUs per node, and the maximum MPI tasks per node) together with the CPUs and GPUs this step requests per task. In particular, on GPU-enabled builds where there is typically one task per GPU, the stride is the number of GPUs per node rather than the number of CPU cores per node.

This updates MPAS-Ocean namelist options directly. For Omega runs, OceanModelStep maps these MPAS options to Omega yaml config options.

Parameters:

config_model (str, optional) – If config options are available for multiple models, the model that the config options are from.