/* Compilation: gcc -o exam2 exam2.c -lpthread */ // Параллельное умножение квадратной матрицы А размерностью NxN // на вектор B, результат - вектор C. // Создается nt потоков, каждый из которых умножает ленту матрицы // A размерностью (N/nt)xN на вектор B. // Умножение повторяется nc раз при неизменной матрице A и // различных векторах B. #include <stdlib.h> #include <unistd.h> #include <stdio.h> #include <string.h> #include <signal.h> #define _REENTRANT #include <pthread.h> #include <sched.h> typedef struct { pthread_t tid; int first; int last; } ThreadRecord; int N; // Размер матрицы A и векторов B, C double *A, *B, *C; int done=0; ThreadRecord *threads; // Массив индексов границ лент матрицы pthread_barrier_t barr1, barr2; void * mysolver (void *arg_p) { ThreadRecord *thr; char numbs[] = {'1','2','3','4','5','6','7','8','9','0'}; int i, j; thr = (ThreadRecord *)arg_p; while ( !done ) { pthread_barrier_wait(&barr1); write (1, numbs, thr->tid); write (1, "\n", 1); for (i=thr->first; i<=thr->last; i++) { C[i] = 0; for (j=0; j<=N; j++) C[i] += A[N*i+j]*B[j]; }; pthread_barrier_wait(&barr2); }; } int main (int argc, char* argv[]) { int nt; // Кол-во расчетных потоков int nc; // Кол-во циклов int i, j, k; pthread_attr_t pattr; if (argc != 4) { fprintf(stderr, "Исп: %s размерность #потоков #циклов\n", argv[0]); exit (1); }; N = atoi(argv[1]); nt = atoi(argv[2]); nc = atoi(argv[3]); if ( N%nt ) { fprintf(stderr, "Размерность д.б. кратна кол-ву потоков\n", argv[0]); exit (2); }; // Выделение памяти A = (double *) malloc( sizeof(double)*N*N ); B = (double *) malloc( sizeof(double)*N ); C = (double *) malloc( sizeof(double)*N ); // Инициализация A и B случ. значениями srand(time(NULL)); for (i=0; i<N; i++) { B[i] = (double) rand(); for (j=0; j<N; j++) A[i*N+j] = (double) rand(); }; pthread_attr_init (&pattr); pthread_attr_setscope (&pattr, PTHREAD_SCOPE_SYSTEM); pthread_attr_setdetachstate (&pattr,PTHREAD_CREATE_JOINABLE); threads = (ThreadRecord *) calloc (nt, sizeof(ThreadRecord)); pthread_barrier_init(&barr1, NULL, nt+1); pthread_barrier_init(&barr2, NULL, nt+1); j = N/nt; for (i=0; i<nt; i++) { threads[i].first = j*i; // Индекс нач. строки А для потока threads[i].last = j*(i+1)-1; // Индекс конечн. строки А для потока if ( pthread_create (&(threads[i].tid), &pattr, mysolver, (void *) &(threads[i])) ) perror("pthread_create"); }; pthread_barrier_wait(&barr1); // Старт расчетов for (k=1; k<nc; k++) { pthread_barrier_wait(&barr2); write(1, "---------\n", 10); for (j=0; j<N; j++) B[j] = (double) rand(); pthread_barrier_wait(&barr1); }; done = 1; exit (0); }