CUDA C 任意矩阵相乘

这篇博客探讨了在CUDA C中实现任意矩阵相乘的问题,重点在于处理矩阵边界和线程块尺寸不匹配的问题。作者建议通过将数据矩阵补全到能够整除32*32的块来简化问题,并提供了相应的补全、初始化和计算矩阵的CUDA C代码示例。
摘要由CSDN通过智能技术生成
好久没来了,继续学习!留下脚印,留下回忆!目前在深圳实习!
/*****************************
*A=M*I,B=I*N
*求矩阵C=A*B,无论怎么看,求任意矩阵内积,即使只用一个grid,在边界上也存在很多问题。
*最主要的就是在边界上,如果block尺寸大于划分的数据小矩阵的尺寸,那么只有部分线程使用,这倒没有问题,
*关键在于,对于部分线程块而言,线程块与矩阵数据块大小不同,就不能做到一个线程取一个数,逻辑就有问题了
*NOTE:考虑这么多,想想最简单的解决办法,还是将数据矩阵补齐为能整除32*32块
*memset()函数因为是按字节填充,所以只适合清零
******************************/

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<cuda_runtime.h>
#include<device_launch_parameters.h>
using namespace std;

#define M 33
#define N 33
#define I 33
#define block_size 32

#define size_a M*I*sizeof(float)
#define size_b I*N*sizeof(float)
#define size_c M*N*sizeof(float)

__global__ void matrix_kernel(float *d_a,float *d_b,float *d_c,int wA,int wB);//wA,wB是A,B矩阵的宽度,也就是列数
void showResult(float *c,int h,int Kw,int w);
void init(float *a,int h,int Kw,int w,int value);//Kw为扩充后矩阵宽度,而w为实际矩阵宽度
//考虑将矩阵补齐,就是行列都变成32*32块的整数倍,函数功能是补齐并初始化,还要分配空间
float* supply(int h,int w,int value);//参数为行,宽,就可以得到扩充后的大小,value为需要初始化的值
int main()
{

	float *h_a,*h_b,*h_c;//声明主机指针
	float *d_a,*d_b,*d_c;//声明设备指针

	//在考虑补齐的时候,实际行数列数都发生了变化,记住最后的矩阵大小为row_c*col_c,但是实际矩阵大小为M*N即可
	int row_a=32*((M+32-1)/32);//补齐后a的行列,但是从[32*M/32,M)为初始化数值,而[M,row_a)则应初始化为零
	int col_a=32*((I+32-1)/32);//[I,col_a)初始化为零
	cout<<"a矩阵扩充后的行列分别为:"<<row_a<<" "&
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值