chundoong-lab-ta/SHPC2022/hw2/matmul/matmul.c

58 lines
1.1 KiB
C
Raw Normal View History

2022-10-04 13:56:31 +09:00
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
struct thread_arg {
const float *A;
const float *B;
float *C;
int M;
int N;
int K;
int num_threads;
int rank; /* id of this thread */
} args[256];
static pthread_t threads[256];
static void* matmul_kernel(void *arg) {
/*
TODO: FILL IN HERE
*/
return NULL;
}
void matmul(const float *A, const float *B, float *C, int M, int N, int K, int num_threads) {
if (num_threads > 256) {
fprintf(stderr, "num_threads must be <= 256\n");
exit(EXIT_FAILURE);
}
int err;
for (int t = 0; t < num_threads; ++t) {
args[t].A = A;
args[t].B = B;
args[t].C = C;
args[t].M = M;
args[t].N = N;
args[t].K = K;
args[t].num_threads = num_threads;
args[t].rank = t;
err = pthread_create(&threads[t], NULL, matmul_kernel, (void*)&args[t]);
if (err) {
printf("pthread_create(%d) failed with err %d\n", t, err);
exit(EXIT_FAILURE);
}
}
for (int t = 0; t < num_threads; ++t) {
err = pthread_join(threads[t], NULL);
if (err) {
printf("pthread_join(%d) failed with err %d\n", t, err);
exit(EXIT_FAILURE);
}
}
}