CUDA教程之——共享存储器(2)-矩阵与自身转置相乘

最新推荐文章于 2024-08-12 01:01:56 发布

豆-Metcalf

最新推荐文章于 2024-08-12 01:01:56 发布

阅读量1.5k

点赞数

分类专栏： CUDA 文章标签： cuda

本文链接：https://blog.csdn.net/u010837794/article/details/53265100

版权

CUDA 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

未经优化的核函数：

__global__ void simpleMultiply(float* a,float* c,int M){
int row=blockIdx.y*blockDim.y+threadIdx.y;
int col=blockIdx.x*blockDim.x+threadIdx.x;
float sum=0.0f;
for(int i=0;i<TILE_DIM;i++){
    sum+=a[row*TILE_DIM+i]*a[col*TILE_DIM+i];
}
c[row*M+col]=sum;
}

优化后的核函数：

__global__ void coalescedMultiply(float* a,float* c,int M){
__shared__ float aTile[TILE_DIM][TILE_DIM],
                                 transposedTile[TILE_DIM][TILE_DIM];
int row=blockIdx.y*blockDim.y+threadIdx.y;
int col=blockIdx.x*blockDim.x+threadIdx.x;
float sum=0.0f;
aTile[threadIdx.y][threadIdx.x]=a[row*TILE_DIM+threadIdx.x];
transposedTile[threadIdx.x][threadIdx.y]=
        a[(blockIdx.x*blockDim.x+threadIdx.y)*TILE_DIM+
        threadIdx.x];
__syncthreads();
for(int i=0;i<TILE_DIM;i++){
        sum+=aTile[threadIdx.y][i]*transposeTile[i][threadIdx.x];
}
c[row*M+col]=sum;}