稀疏矩阵 part 4
2024-08-27 14:50:07
▶ 各种稀疏矩阵数据结构下 y(n,1) = A(n,m) * x(m,1) 的实现，GPU版本
● MAT 乘法
__global__ void dotGPU(const MAT *a, const MAT *x, MAT *y)
{
int id = blockIdx.x * blockDim.x + threadIdx.x;
if (id < a->row)
{
format sum = ;
for (int i = ; i < a->col; i++)
sum += a->data[id * a->col + i] * x->data[i];
y->data[id] = sum;
}
if (id == )
{
y->row = a->row;
y->col = x->col;
COUNT_MAT(y);
}
return;
}
● CSR 乘法
__global__ void dotGPU(const CSR *a, const MAT *x, MAT *y)
{
int id = blockIdx.x * blockDim.x + threadIdx.x;
if (id < a->row)
{
format sum = ;
for (int j = a->ptr[id]; j < a->ptr[id + ]; j++)
sum += a->data[j] * x->data[a->index[j]];
y->data[id] = sum;
}
if (id == )
{
y->row = a->row;
y->col = x->col;
COUNT_MAT(y);
}
return;
}
● ELL 乘法
__global__ void dotGPU(const ELL *a, const MAT *x, MAT *y)
{
int id = blockIdx.x * blockDim.x + threadIdx.x;
if (id < a->col)
{
format sum = ;
for (int j = ; j < a->row; j++)
sum += a->data[id + j * a->col] * (a->index[id + j * a->col] < ? : x->data[a->index[id + j * a->col]]);
y->data[id] = sum;
}
if (id == )
{
y->row = a->col;
y->col = x->col;
COUNT_MAT(y);
}
return;
}
● COO 乘法
__global__ void dotGPU(const ELL *a, const MAT *x, MAT *y)// GPU ELL乘法
{
int id = blockIdx.x * blockDim.x + threadIdx.x;
if (id < a->col)
{
format sum = ;
for (int j = ; j < a->row; j++)
sum += a->data[id + j * a->col] * (a->index[id + j * a->col] < ? : x->data[a->index[id + j * a->col]]);
y->data[id] = sum;
}
if (id == )
{
y->row = a->col;
y->col = x->col;
COUNT_MAT(y);
}
return;
}
● DIA 乘法，留坑
最新文章
- PageImpl是不是有问题？
- F - To the Max
- python开发_mysqldb安装
- XunSearch（讯搜）的使用教程步骤
- s1.charAt(x)=='a'
- 我的github
- MVC通俗演义系列开篇
- Android Fragment详解(三)： 实现Fragment的界面
- 除trigger()方法外的jquery手动触发事件
- Linux入门练习
- AES解密后多了\0
- functiontools模块
- JavaFx之不通过全局静态变量进行窗体通信
- python3版本中的zip函数
- setattr
- python中关于if-else使用性能的一点感悟
- VS 清除编译产生的临时文件、文件夹
- Windows下本机简易监控系统搭建（Telegraf+Influxdb+Grafana）
- 520的信心赛——点点玩deeeep
- java的开发主要以http为基础