SIMD инструкции процессора

#include <iostream>
#include <immintrin.h> 
 
const int N = 4096; 
 
 
float rand_float() {
    return static_cast<float>(rand()) / RAND_MAX;
}
 
 
void matrix_multiply_simd(float** A, float** B, float** C) {
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            __m128 result = _mm_setzero_ps();  
            for (int k = 0; k < N; k += 4) {
                __m128 a = _mm_load_ps(&A[i][k]);
                __m128 b = _mm_load_ps(&B[k][j]);
                result = _mm_add_ps(result, _mm_mul_ps(a, b));  
            }
            _mm_store_ps(&C[i][j], result);  
        }
    }
}
 
 
void matrix_multiply_scalar(float** A, float** B, float** C) {
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            C[i][j] = 0.0;
            for (int k = 0; k < N; k++) {
                C[i][j] += A[i][k] * B[k][j]; 
            }
        }
    }
}
 
int main() {
    float** A = new float* [N];
    float** B = new float* [N];
    float** C_simd = new float* [N];
    float** C_scalar = new float* [N];
 
    for (int i = 0; i < N; i++) {
        A[i] = new float[N];
        B[i] = new float[N];
        C_simd[i] = new float[N];
        C_scalar[i] = new float[N];
    }
 
    
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            A[i][j] = rand_float();
            B[i][j] = rand_float();
        }
    }
 
   
    matrix_multiply_simd(A, B, C_simd);
 
   
    matrix_multiply_scalar(A, B, C_scalar);
 
   
    bool result_match = true;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            if (C_simd[i][j] != C_scalar[i][j]) {
                result_match = false;
                break;
            }
        }
        if (!result_match) {
            break;
        }
    }
 
    if (result_match) {
        std::cout << "Векторное умножение работает правильно." << std::endl;
    }
    else {
        std::cout << "Векторное умножение работает неправильно." << std::endl;
    }
 
    for (int i = 0; i < N; i++) {
        delete[] A[i];
        delete[] B[i];
        delete[] C_simd[i];
        delete[] C_scalar[i];
    }
    delete[] A;
    delete[] B;
    delete[] C_simd;
    delete[] C_scalar;
 
    std::cin.get();
 
    return 0;
}