@specs-feup/clava/src-api/clava/mpi/MpiUtils.ts

A C/C++ source-to-source compiler written in Typescript

import { BuiltinType, Type } from "../../Joinpoints.js";

/**
 * Utility methods related to MPI.
 *
 */
export default class MpiUtils {
  static VAR_NUM_TASKS = "mpi_num_tasks";
  static VAR_NUM_WORKERS = "mpi_num_workers";
  static VAR_RANK = "mpi_rank";
  static _VAR_MPI_STATUS = "mpi_status";

  static getMpiType($type: Type) {
    // Reference: https://msdn.microsoft.com/en-us/library/windows/desktop/dn473290(v=vs.85).aspx

    if ($type instanceof BuiltinType) {
      return "MPI_" + $type.code.toUpperCase().replace(" ", "_");
    }

    throw "Not implemented for type '" + $type.astName + "'";
  }
}