Communication graph and Gantt chart

Mellich commented 1 year ago

This PR does the following changes:

Extend MPIEvent with an end time to allow the generation of a Gantt chart (#3)
Adds functionality to generate a communication graph from a merged tape
Primitive plotting functions for Gantt charts and arrows to reflect communication between ranks

Mellich commented 1 year ago

I also added support for #1 with the function plot_sequence_merged. It is possible to output the graph directly in the terminal.

carstenbauer commented 1 year ago

Looks nice! I was first slightly confused that there are no MPI_Recvs. :)

BTW, there is a dropmpiprefix function (or similar) that drops the "MPI_" part from the function names. Should look a lot cleaner with this.

     ┌──────┐          ┌──────┐          ┌──────┐          ┌──────┐          ┌──────┐
     │Rank_0│          │Rank_1│          │Rank_2│          │Rank_4│          │Rank_3│
     └──┬───┘          └──┬───┘          └──┬───┘          └──┬───┘          └──┬───┘
        │     MPI_Send    │                 │                 │                 │
        │ ────────────────>                 │                 │                 │
        │                 │                 │                 │                 │
        │              MPI_Send             │                 │                 │
        │ ──────────────────────────────────>                 │                 │
        │                 │                 │                 │                 │
        │     MPI_Send    │                 │                 │                 │
        │ ────────────────>                 │                 │                 │
        │                 │                 │                 │                 │
        │                 │     MPI_Send    │                 │                 │
        │ ────────────────────────────────────────────────────>                 │
        │                 │                 │                 │                 │
        │              MPI_Send             │                 │                 │
        │ ──────────────────────────────────>                 │                 │
        │                 │                 │                 │                 │
        │                 │     MPI_Send    │                 │                 │
        │ ────────────────────────────────────────────────────>                 │
        │                 │                 │                 │                 │
        │                 │              MPI_Send             │                 │
        │ ──────────────────────────────────────────────────────────────────────>
        │                 │                 │                 │                 │
        │                 │              MPI_Send             │                 │
        │ ──────────────────────────────────────────────────────────────────────>
        │                 │                 │                 │                 │
        │                 │     MPI_Send    │                 │                 │
        │ ────────────────────────────────────────────────────>                 │
        │                 │                 │                 │                 │
        │     MPI_Send    │                 │                 │                 │
        │ ────────────────>                 │                 │                 │
        │                 │                 │                 │                 │
        │                 │              MPI_Send             │                 │
        │ ──────────────────────────────────────────────────────────────────────>
        │                 │                 │                 │                 │
        │              MPI_Send             │                 │                 │
        │ ──────────────────────────────────>                 │                 │
        │                 │                 │                 │                 │
        │     MPI_Send    │                 │                 │                 │
        │ <────────────────                 │                 │                 │
        │                 │                 │                 │                 │
        │              MPI_Send             │                 │                 │
        │ <──────────────────────────────────                 │                 │
        │                 │                 │                 │                 │
        │                 │              MPI_Send             │                 │
        │ <──────────────────────────────────────────────────────────────────────
        │                 │                 │                 │                 │
        │                 │     MPI_Send    │                 │                 │
        │ <────────────────────────────────────────────────────                 │
     ┌──┴───┐          ┌──┴───┐          ┌──┴───┐          ┌──┴───┐          ┌──┴───┐
     │Rank_0│          │Rank_1│          │Rank_2│          │Rank_4│          │Rank_3│
     └──────┘          └──────┘          └──────┘          └──────┘          └──────┘

pc2 / MPITape.jl

Communication graph and Gantt chart #6