cuda 类模板以及全局内存的使用例子

#include "cuda_runtime.h"
#include "device_launch_parameters.h"

#include <stdio.h>
#include<stdlib.h>
#define MAX_VALUES 1024
template<class T>
class myValues{
	T values[MAX_VALUES];
public:
	__device__ myValues(T clear){}
	__device__ void setValue(int Idx,T value){
	values[Idx] = value;
	}
	__device__ void putToMemory(int Idx,T* valueLocation){
		if(Idx<MAX_VALUES){
		valueLocation[Idx] = values[Idx]+Idx;
		}
	}
};

template<class T>
void __global__ useValues(T* memBuffer){
	myValues<T> myLocation(0);
	int tid = blockIdx.x*blockDim.x+threadIdx.x;
	myLocation.setValue(tid,1);
	myLocation.putToMemory(tid,memBuffer);
}

static int* data_host=(int*)malloc(MAX_VALUES*sizeof(int));
__device__ void* buffer;

int main(){
	cudaMalloc((void**)&buffer,MAX_VALUES*sizeof(int));
useValues<int><<<2,512>>>((int*)buffer);
cudaMemcpy(data_host,buffer,MAX_VALUES*sizeof(int),cudaMemcpyDeviceToHost);
for(int i=0;i<MAX_VALUES;i++){
	printf("%d\n",data_host[i]);
	}
system("pause");
return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值