#!/bin/sh
#
# SPDX-License-Identifier: MIT
#
# SPDX-FileCopyrightText: 2023 Olivier Dion <odion@efficios.com>

TRACE_OUTPUT=traces

lttng create --output $TRACE_OUTPUT

# Enable all MPI event types.
lttng enable-event --userspace 'mpi:*'

# Add MPI rank as a application context to every event.
lttng add-context --userspace --type '$app.MPI:rank'

lttng start

if command -v mpirun;
then
    mpirun --n 4 ./run-test-mpi 100
else
    srun -p cray --ntasks=4 ./run-test-mpi 1000
fi

lttng destroy

babeltrace2 $TRACE_OUTPUT