Skip to content

HIP: test for GPU-aware MPI with ucx_info

Henri Menkerequested to merge
hip into main

Description

If OpenMPI is not compiled explicitly with --with-cuda or --with-rocm, then the compile-time constants in <mpi-ext.h> are not available. However, the GPU-aware support for OpenMPI does not actually come from OpenMPI itself, but from the underlying UCX. Therefore we can also test if UCX was compiled with --with-cuda or --with-rocm. This can be done by running ucx_info -b and inspecting the line with the UCX_CONFIGURE_FLAGS.

TODO: Needs to be tested properly.

News snippet

Checklist

  • I have checked that my code follows the Octopus coding standards
  • I have added tests for all the new features added in this request.

Merge request reports