ompi-musa

ompi-musa is Moore Threads' MUSA adaptation of Open MPI, an open source implementation of the Message Passing Interface (MPI) specification. It is intended for high-performance computing applications running on Moore Threads MUSA GPUs.

The project keeps the Open MPI MCA architecture and MPI interfaces while adding MUSA accelerator support for device-memory-aware communication. The upstream Open MPI documentation remains applicable unless a MUSA-specific section is called out below.

Version and source

• Project release: v6.1.0a1-musa1
• Upstream baseline: Open MPI 6.1.0a1
• Upstream baseline commit: 625232007beb2214f2fa7460df8736a2d53b34f9
• MUSA integration branch: musa-aware-ompi-opt

MUSA support

The MUSA adaptation includes:

• a MUSA accelerator component using the MUSA driver API;
• MUSA device-memory detection, asynchronous copies, streams, and events;
• MUSA-aware MPI extension support, including MPIX_Query_musa_support();
• device-memory communication through shared-memory and IPC paths; and
• build-time detection of the MUSA SDK and MUSA-enabled MCA components.

Building with MUSA

The MUSA SDK and the Open MPI build dependencies must be installed before configuration. Adjust the SDK and library paths for the target system.

git clone <repository-url> ompi-musa
cd ompi-musa
git checkout musa-aware-ompi-opt
git submodule update --init --recursive

perl autogen.pl
mkdir build
cd build
../configure \
    --with-musa=/usr/local/musa \
    --with-musa-libdir=/usr/lib/x86_64-linux-gnu \
    --enable-mpiext=musa \
    --with-pmix=internal \
    --with-prrte=internal \
    --prefix=/usr/local
make -j"$(nproc)"
make check
make install

After installation, verify that the MUSA components are available:

ompi_info --parsable --all | grep -i musa

Official Open MPI documentation

The Open MPI documentation can be viewed online at https://docs.open-mpi.org/. A self-contained copy can be generated in official distribution tarballs under docs/_build/html/index.html.

The source for the documentation is in the docs directory. Instructions for building it locally are available at https://docs.open-mpi.org/en/main/developers/prerequisites.html.

License

The MUSA adaptations and Moore Threads modifications in this distribution are licensed under the BSD 3-Clause License. Original Open MPI code and other third-party components remain under their respective licenses. See LICENSE for the complete notices and terms.