Skip to content

Before the run resource estimator for CPU (OpenMP) and GPU (CUDA) #325

Description

@kaanolgu

Provide a small, self-contained test that tells user before launching a large case how much memory it will need per GPU for the CUDA backend, and per rank (host RAM) for the OpenMP CPU backend, so they don't queue a job that would end up being OOM

Proposed behaviour

  • One command, reading the real input file (grid, BCs, schemes, decomposition).
  • Reports the exact allocator footprint for a sweep of device/rank counts.
  • Calibrates the harder to predict term by building a couple of small grids and measuring:
    • CUDA: the cuFFT/cuFFTMp + context workspace
    • OpenMP CPU: the 2decomp/FFT plans + per-rank host buffers, against host RAM.
  • Flags configurations that exceed device/host memory and prints the minimum device/rank count required.

Todo list:

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions