cpu矩阵分块乘法C语言代码

以下是一个简单的矩阵分块乘法的 C 代码实现：

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

#define N 1024
#define B 32

// 矩阵相乘函数
void matrix_multiply(int n, double **A, double **B, double **C)
{
    int i, j, k, ii, jj, kk;
    double sum;

    for (i = 0; i < n; i++)
        for (j = 0; j < n; j++)
            C[i][j] = 0.0;

    for (kk = 0; kk < n; kk += B)
        for (jj = 0; jj < n; jj += B)
            for (i = 0; i < n; i++)
                for (j = jj; j < jj + B; j++)
                {
                    sum = C[i][j];
                    for (k = kk; k < kk + B; k++)
                        sum += A[i][k] * B[k][j];
                    C[i][j] = sum;
                }
}

int main()
{
    int i, j;
    double **A, **B, **C;

    // 动态分配内存
    A = (double **) malloc(N * sizeof(double *));
    B = (double **) malloc(N * sizeof(double *));
    C = (double **) malloc(N * sizeof(double *));
    for (i = 0; i < N; i++)
    {
        A[i] = (double *) malloc(N * sizeof(double));
        B[i] = (double *) malloc(N * sizeof(double));
        C[i] = (double *) malloc(N * sizeof(double));
    }

    // 初始化矩阵
    for (i = 0; i < N; i++)
        for (j = 0; j < N; j++)
        {
            A[i][j] = i + j;
            B[i][j] = i - j;
        }

    // 执行矩阵相乘
    matrix_multiply(N, A, B, C);

    // 输出结果
    for (i = 0; i < N; i++)
    {
        for (j = 0; j < N; j++)
            printf("%f ", C[i][j]);
        printf("\n");
    }

    // 释放内存
    for (i = 0; i < N; i++)
    {
        free(A[i]);
        free(B[i]);
        free(C[i]);
    }
    free(A);
    free(B);
    free(C);

    return 0;
}

在这个实现中，我们使用了一个 BxB 的块大小来分割输入矩阵，并使用了四个 for 循环来完成矩阵相乘的计算。在实践中，可以调整块大小以获得更好的性能。