cuda笔记-第一个cuda程序

﹏ヽ暗。殇╰゛Y 2022-12-27 08:29 211阅读 0赞

这里先说明下一些基本概念：

释放GPU中的内存cudaFree()

CUDA函数的定义：

\_\_global\_\_：定义在GPU上，可以在CPU上调用的函数；

\_\_device\_\_：定义在GPU上，由GPU调用函数；

\_\_host\_\_：在CPU上定义的函数，一般与\_\_device\_\_一起用

在GPU上开辟空间：cudaMalloc(\*\*devPtr, byte\_size)

如：

int *gpu_int;
    cudaMalloc((void**)&gpu_int, sizeof(int))

GPU上数组的初始化cudaMemset(\*devptr, value, byte\_size)

GPU、CPU参数传递cudaMemcpy(\*dst, \*src, byte\_size, 类型)

其中这个类型包括：

CPU2CPU：cudaMemcpyHostToHost

CPU2GPU：cudaMemcpyHostToDevice

GPU2CPU：cudaMemcpyDeviceToHost

GPU2GPU：cudaMemcpyDeviceToDevice

源码如下：

#include "cuda_runtime.h"
    #include "device_launch_parameters.h"
    #include <stdio.h>
    
    using namespace std;
    
    __device__ int add_one(int a) {
    
    	return a + 1;
    }
    
    __global__ void show(int *a) {
    
    	for (int i = 0; i < 10; i++) {
    
    		//a[i] = add_one(a[i]);
    		printf(" %d", a[i]);
    	}
    	printf("\n");
    }
    
    __global__ void changeValue(int *a) {
    
    	for (int i = 0; i < 10; i++) {
    
    		a[i] = 100;
    	}
    }
    
    int main() {
    
    	int cpu_int[10] = {10, 10, 10, 10, 10, 10, 10, 10, 10, 10};
    	int *gpu_int;
    
    	//在GPU上分配空间
    	cudaMalloc((void**)&gpu_int, 10 * sizeof(int));
    	show << <1, 1 >> > (gpu_int);
    
    	//初始化其值
    	cudaMemset(gpu_int, 0, 10 * sizeof(int));
    	show<< <1, 1 >> > (gpu_int);
    
    	//将cpu_int赋值给gpu_int
    	cudaMemcpy(gpu_int, cpu_int, 10 * sizeof(int), cudaMemcpyHostToDevice);
    	show << <1, 1 >> > (gpu_int);
    
    	//改变gpu_int的值
    	changeValue << <1, 1 >> >(gpu_int);
    	show << <1, 1 >> > (gpu_int);
    
    	//将gpu_int的值赋值到cpu_int
    	cudaMemcpy(cpu_int, gpu_int, 10 * sizeof(int), cudaMemcpyDeviceToHost);
    	printf("----------华丽的分割线----------\n");
    	for (int i = 0; i < 10; i++) {
    
    
    		printf(" %d", cpu_int[i]);
    	}
    
    
    	//释放gpu_int的空间
    	cudaFree(gpu_int);
    
    
    	getchar();
    	return 0;
    }

程序运行截图如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNzg0NDI3NjE_size_16_color_FFFFFF_t_70][]

源码打包地址如下：

[https://github.com/fengfanchen/CAndCPP/tree/master/HelloCuda][https_github.com_fengfanchen_CAndCPP_tree_master_HelloCuda]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNzg0NDI3NjE_size_16_color_FFFFFF_t_70]: /images/20221120/97d93e51b53a4e43b37328770e838a01.png
[https_github.com_fengfanchen_CAndCPP_tree_master_HelloCuda]: https://github.com/fengfanchen/CAndCPP/tree/master/HelloCuda