尝试对下面这段矩阵乘法代码使用向量化技术进行优化。 void gemm...

Question

尝试对下面这段矩阵乘法代码使用向量化技术进行优化。

void gemm(int M, int N, int K, const float* a, const float* b, float* c) {
    for (int i = 0 i < M ++i) {
        for (int j = 0 j < N ++j) {
            float sum = 0
            for (int kk = 0 kk < K ++k) {
                sum += a[i * K + kk] * b[kk * N + j]
            }
            c[i * N + j] = sum
        }
    }
}

答题说明：该题为手动判卷，答案只要写对即可，不用严格满足字符比对。