MPIVersion.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Copyright (c) 2016-2018 Intel, Inc.  All rights reserved.
# $COPYRIGHT$
#
# Additional copyrights may follow
#
# $HEADER$
#

import os
import sys
from BaseMTTUtility import *
import shlex

## @addtogroup Utilities
# @{
# @section MPIVersion
# Identify the name and version of MPI in-use
# @}
class MPIVersion(BaseMTTUtility):
    def __init__(self):
        BaseMTTUtility.__init__(self)
        self.options = {}
        return

    def print_name(self):
        return "MPIVersion"

    def print_options(self, testDef, prefix):
        lines = testDef.printOptions(self.options)
        for line in lines:
            print(prefix + line)
        return

    def execute(self, log, testDef):

        version_str = self.get_version_string(testDef)

        if version_str is None:
            log['name'] = 'None'
            log['version'] = 'Unknown'
            return

        name = None
        version = None

        # Open MPI
        # Example Output:
        # Open MPI v1.10.2, package: Open MPI abuild@ip-172-31-24-182.us-west-2.compute.internal Distribution, ident: 1.10.2, repo rev: v1.10.1-145-g799148f, Jan 21, 2016
        if 'Open MPI' in version_str:
            name = 'Open MPI'
            version = version_str.split('Open MPI v')[1].split(', ')[0]

        # MVAPICH2
        # Example Output:
        # MVAPICH2 Version      : 2.1
        # MVAPICH2 Release date : Fri Apr 03 20:00:00 EDT 2015
        # MVAPICH2 Device       : ch3:mrail
        # MVAPICH2 configure    : --prefix=/opt/ohpc/pub/mpi/mvapich2-gnu-ohpc/2.1 --enable-cxx --enable-g=dbg --with-device=ch3:mrail --enable-fast=O3
        # MVAPICH2 CC           : gcc    -g -O3
        # MVAPICH2 CXX          : g++   -g -O3
        # MVAPICH2 F77          : gfortran -L/lib -L/lib   -g -O3
        # MVAPICH2 FC           : gfortran   -g -O3
        elif 'MVAPICH2' in version_str:
            name = 'MVAPICH2'
            version = version_str.split('MVAPICH2 Version')[1].split(':')[1].split("'")[0].split("\\t")[1]
        # Intel MPI
        # Example Output:
        # Intel(R) MPI Library 5.1.3 for Linux* OS
        elif 'Intel' in version_str:
            name = 'Intel MPI'
            version = version_str.split('Intel(R) MPI Library ')[1].split(' ')[0]

        elif 'CRAY MPICH' in version_str:
            name = 'CRAY MPICH'
            version = version_str.split('CRAY MPICH version ')[1].split(' ')[0]

        # record the result
        log['name'] = str(name)
        log['version'] = str(version)
        return

    def get_version_string(self, testDef):
        os.chdir(testDef.options['scratchdir'])

        try:
            fh = open("mpi_get_version.c", "r")
            fh.close()
        except IOError:
            fh = open("mpi_get_version.c", "w")
            fh.write("""
/* This program is automatically generated by MPIVersion.py
 * of MPI Testing Tool (MTT). Any changes you make here may
 * get lost!
 * Copyrights and licenses of this file are the same as for the MTT.
 */
#include <mpi.h>
#include <stdio.h>
int main(int argc, char **argv) {
    char version[3000];
    int resultlen;
    MPI_Get_library_version(version, &resultlen);
    printf("%s\\n", version);
    return 0;
}""")
            fh.close()
        status, _, _, _ = testDef.execmd.execute(None, shlex.split('mpicc -o mpi_get_version mpi_get_version.c'), testDef)
        if 0 != status:
            status, _, _, _ = testDef.execmd.execute(None, shlex.split('cc -o mpi_get_version mpi_get_version.c'), testDef)
            if 0 != status:
                os.chdir("..")
                return None

        status, stdout, _, _ = testDef.execmd.execute(None, shlex.split('sh -c "mpiexec ./mpi_get_version |uniq -c"'), testDef)
        if 0 != status:
            status, stdout, _, _ = testDef.execmd.execute(None, shlex.split('sh -c "aprun ./mpi_get_version |uniq -c"'), testDef)
            if 0 != status:
                status, stdout, _, _ = testDef.execmd.execute(None, shlex.split('sh -c "./mpi_get_version |uniq -c"'), testDef)
                if 0 != status:
                    os.chdir("..")
                    return None

        os.chdir("..")

        return "\n".join(stdout)