c - Pthreads矩阵乘法错误

我想在我现有的串行矩阵乘法代码中使用p螺纹。我的目标是使用pthreads获得更好的执行时间，只是为了实现加速。但在那一点上我被卡住了。我原来的串行代码，工作得很好，我在15秒内完成了1000x1000平方矩阵乘法。但是当我执行当前的pthreads程序时，我会遇到一个分段错误。这是我的代码：

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <assert.h>

int SIZE, NTHREADS;
int **A, **B, **C;

void init()
{
    int i, j;

    A = (int**)malloc(SIZE * sizeof(int *));
    for(i = 0; i < SIZE; i++)
        A[i] = malloc(SIZE * sizeof(int));

    B = (int**)malloc(SIZE * sizeof(int *));
    for(i = 0; i < SIZE; i++)
        B[i] = malloc(SIZE * sizeof(int));

    C = (int**)malloc(SIZE * sizeof(int *));
    for(i = 0; i < SIZE; i++)
        C[i] = malloc(SIZE * sizeof(int));

    srand(time(NULL));

    for(i = 0; i < SIZE; i++) {
        for(j = 0; j < SIZE; j++) {
            A[i][j] = rand()%100;
            B[i][j] = rand()%100;
        }
    }
}

void mm(int tid)
{
    int i, j, k;
    int start = tid * SIZE/NTHREADS;
    int end = (tid+1) * (SIZE/NTHREADS) - 1;

    for(i = start; i <= end; i++) {
        for(j = 0; j < SIZE; j++) {
            C[i][j] = 0;
            for(k = 0; k < SIZE; k++) {
                C[i][j] += A[i][k] * B[k][j];
            }
        }
    }
}

void *worker(void *arg)
{
    int tid = *((int *) arg);
    mm(tid);
}

int main(int argc, char* argv[])
{
    pthread_t* threads;
    int rc, i;

    if(argc != 3)
    {
        printf("Usage: %s <size_of_square_matrix> <number_of_threads>\n", argv[0]);
        exit(1);
    }

    SIZE = atoi(argv[1]);
    NTHREADS = atoi(argv[2]);
    init();
    threads = (pthread_t*)malloc(NTHREADS * sizeof(pthread_t));

    clock_t begin, end;
    double time_spent;


    begin = clock();

    for(i = 0; i < NTHREADS; i++) {
        rc = pthread_create(&threads[i], NULL, worker, (void *)i);
        assert(rc == 0);
    }

    for(i = 0; i < NTHREADS; i++) {
        rc = pthread_join(threads[i], NULL);
        assert(rc == 0);
    }

    end = clock();

    time_spent = (double)(end - begin) / CLOCKS_PER_SEC;
    printf("Elapsed time: %.2lf seconds.\n", time_spent);

    for(i = 0; i < SIZE; i++)
        free((void *)A[i]);
    free((void *)A);

    for(i = 0; i < SIZE; i++)
        free((void *)B[i]);
    free((void *)B);

    for(i = 0; i < SIZE; i++)
        free((void *)C[i]);
    free((void *)C);

    free(threads);

    return 0;
}

如果有人能帮助我运行pthreads程序，并实现一些速度提升，我会很高兴。

最佳答案

对于当前代码，应该使用

int tid = (int)arg;

（您的代码有效地将循环计数器视为一个地址，然后在0或0左右取消对地址的引用。这些地址可能无法被您的进程读取和/或无法正确对齐，因此出现seg故障）
上述更改可能会使您的工作正常，但请注意，将int作为void*传递并不是完全正确的。它依赖于sizeof(int) <= sizeof(void*)，这很可能但不能保证是真的。如果您关心这个问题，您可以为传递给每个线程的数据分配内存，也可以传递i的地址并包括同步，以确保在每次pthread_create调用之后等待，直到线程被调度并读取其参数。

关于c - Pthreads矩阵乘法错误，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/15974499/