cuda有那些常用api
一些常用的 CUDA API 如下:
- cudaMalloc:在设备上分配内存
- cudaMemcpy:在设备和主机之间进行内存拷贝
- cudaFree:释放设备上的内存
- cudaMemset:在设备上设置内存值
- cudaGetDeviceCount:获取可用的 CUDA 设备数量
- cudaGetDeviceProperties:获取 CUDA 设备的属性信息
- cudaStreamCreate:创建 CUDA 流
- cudaStreamSynchronize:同步 CUDA 流
- cudaMemcpyAsync:异步进行内存拷贝
- cudaDeviceSynchronize:同步设备上的所有任务
- cudaThreadSynchronize:同步主机和设备之间的所有任务
- cudaEventCreate:创建 CUDA 事件
- cudaEventRecord:记录 CUDA 事件
- cudaEventSynchronize:同步 CUDA 事件
- cudaEventElapsedTime:计算 CUDA 事件之间的时间差
- cudaThreadExit:退出 CUDA 线程
- cudaSetDevice:设置当前活动的 CUDA 设备
- cudaGetLastError:获取最近一次 CUDA 函数调用的错误码
- cudaDeviceReset:重置当前活动的 CUDA 设备
- cudaOccupancyMaxPotentialBlockSize:计算每个 SM 的最大线程块数量和每个线程块的最大线程数量
这只是一些常用的 CUDA API,还有很多其他的 API 可以根据需求进行使用
原文地址: https://www.cveoy.top/t/topic/hThU 著作权归作者所有。请勿转载和采集!