MPI，openMP与pthread的基本demo

MPI demo

编译命令：mpicc -g -Wall -o Hello_mpi Hello_mpi.c

执行命令：mpiexec -n 16 ./Hello_mpi

#include <stdio.h>
#include <string.h>
#include <mpi.h>

const int MAX_STRING = 100;
int main()
{
    char greeting[MAX_STRING];
    int comm_sz;
    int my_rank;
    int i = 1;

    MPI_Init(NULL, NULL);
    MPI_Comm_size(MPI_COMM_WORLD, &comm_sz);
    MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);

    if (my_rank != 0)
    {
        sprintf(greeting, "Greetings from process %d of %d!",my_rank, comm_sz);
        MPI_Send(greeting, strlen(greeting)+1, MPI_CHAR, 0, 0, MPI_COMM_WORLD);
    }else{
        printf("Gteetings from process %d of %d!\n",my_rank, comm_sz);
        for(i = 1; i < comm_sz; i++)
        {
            MPI_Recv(greeting, MAX_STRING, MPI_CHAR, i, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
            printf("%s\n", greeting);
        }
    }

    MPI_Finalize();
    return 0;
}

pthread Demo

编译命令：gcc -g -Wall -o pth_hello pth_hello.c -lpthread

执行命令：./pth <number of threads>

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>

int thread_count;

void* Hello(void* rank);

int main(int argc, char* argv[])
{
    long thread = 0;

    pthread_t* thread_handles;

    thread_count = strtol(argv[1], NULL, 10);

    thread_handles = malloc(thread_count * sizeof(pthread_t));

    for(thread = 0; thread < thread_count; thread++)
    {
        pthread_create(&thread_handles[thread], NULL, Hello, (void*) thread);
    }

    for(thread = 0; thread < thread_count; thread++)
    {
        pthread_join(thread_handles[thread], NULL);
    }

    free(thread_handles);
    return 0;
}

void* Hello(void* rank)
{
    long my_rank = (long) rank;
    printf("Hello from thread %ld of %d\n", my_rank, thread_count);

    return NULL;
}

openMP Demo

编译命令：gcc -g -Wall -fopenmp -o omp_hello omp_hello.c

运行命令：./omp_hello <number of thread>

#include <stdio.h>
#include <stdlib.h>
#include <omp.h>

void Hello(void);

int main(int argc, char* argv[])
{
    int thread_count = strtol(argv[1], NULL, 10);

#   pragma omp parallel num_threads(thread_count)
    Hello();

    return 0;
}

void Hello(void)
{
    int my_rank = omp_get_thread_num();
    int thread_count = omp_get_num_threads();

    printf("Hello from thread %d of %d\n", my_rank, thread_count);
}