GPU编程学习

本文展示了如何在Linux环境下使用C++和CUDA进行混合编程,实现矩阵元素取模运算。通过创建CUDA消息处理器,分配GPU资源,执行核函数,将结果从GPU拷贝回CPU,最后展示原矩阵和取模后的矩阵。
摘要由CSDN通过智能技术生成

1 Linux上C++与CUDA混合编程
完整代码及运算结果
/*
mod_test.cu
*/
#include “cuda_runtime.h”
#include “cublas_v2.h”
#include “device_launch_parameters.h”
#include
#include <stdio.h>

using namespace std;

global void mod_kernel(double* d_M1, double* d_M2, long q, int n) {
// index
int idx = blockIdx.x * blockDim.x + threadIdx.x;
long x;
if (idx < n){
x = d_M1[idx];
// printf(“x: %d\n”, x)
d_M2[idx] = x % q;
}
}

long** mat_mod_cu(long** M1, long rowSize, long colSize, long q, cublasHandle_t cuHandle) {
// 1.定义结果矩阵,用于返回
long** M2 = new long* [rowSize];
for (long i = 0; i < rowSize; i++)
M2[i] = new long[colSize];
// 2.分配CPU资源
double* h_M1 = (double*)malloc(rowSize * colSize * sizeof(double));
double* h_M2 = (double*)malloc

  • 9
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值