CUDA之矩阵乘法——TILE&sharedmemory

最新推荐文章于 2024-04-19 13:12:54 发布

Sumujingling

最新推荐文章于 2024-04-19 13:12:54 发布

阅读量3.2k

点赞数

分类专栏： CUDA

本文链接：https://blog.csdn.net/Sumujingling/article/details/51496236

版权

本文介绍了如何在CUDA中通过TILE分块和shared memory技术实现矩阵乘法，通过代码示例展示并行计算过程，并验证了结果与MATLAB计算的一致性。

摘要由CSDN通过智能技术生成

CUDA 矩阵乘法

将输入数据分成很多个TILE使用shared memory进行并行计算

矩阵乘法分块计算

代码

#include "device_functions.h"
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>
#include <iostream>
typedef struct {
    int width;
    int height;
    int stride;
    float* elements;
} Matrix;

#define BLOCK_SIZE  16
#define N           3072

__device__ float GetElement(const Matrix A, int row, int col) {
    return A.elements[row * A.stride + col];
}

__device__ void SetElement(Matrix A, int row, int col, float value) {
    A.elements[row * A.stride + col] = value;
}

__device__ Matrix GetSubMatrix(Matrix A, int row, int co