CUDA編程概述

CUDA C 的編程接口主要包括一個對C語言的小擴展集以及一個runtime庫。

核心的語言擴展在于編程模型,就是kernel,block,grid,完整的擴展參閱 C Language Extensions,任何源文件包括了擴展內容就必須用nvcc編譯,如果沒有使用擴展語言,可以在正常的.cpp文件中調用CUDA函數

在編譯流程中引入runtime,它讓host端通過C函數來控制device端內存分配和釋放,兩者內存之間的數據轉換,管理多device系統等。
runtime建立在底層的CUDA驅動程序API之上,驅動程序API也提供了一些接口可以由應用程序訪問,例如CUDA contexts和CUDA module。但是大多數應用程序不使用驅動程序API,因為它們不需要這種額外的控制,并且在使用runtime時,上下文和模塊管理是隱式的,代碼可以更簡潔。

基礎

  1. 文件后綴為.cu 和 .cuh,CUDA 源文件和頭文件,其他的和正常的.cpp、.hpp一致。
  2. host端支持所有的C++語法,但是device端支持部分C++語法

函數類型限定符Function Type Qualifiers

__host__

函數的調用和執行都在host上

__device__

函數的調用和執行都在device上

__global__

函數在host上調用,在device上執行,對于Compute Capability 3.2及以上也可以在device上調用,特性如下

  1. 對應函數返回值必定為為void
  2. 調用__global__函數必須有執行配置(execution configuration),即<<< Dg, Db, Ns, S >>>參數
  3. 對應修飾的函數為異步

變量類型修飾符Variable Type Qualifiers

指明在device中變量的內存位置,在device端的代碼中沒有__device__, __shared__ and __constant__修飾符,則為自動變量,在寄存器register中。

__device__

變量在device中,常常和其他類型修飾符一起使用,如果只有該修飾符

  1. 在global內存空間中
  2. 擁有整個應用的生命周期
  3. 可以被grid中的所有線程訪問,同時還host端還可以通過runtime庫訪問(cudaGetSymbolAddress() / cudaGetSymbolSize() / cudaMemcpyToSymbol() / cudaMemcpyFromSymbol()).
__constant__

可以和__device__修飾符一同使用

  1. 在constant內存空間中
  2. 擁有整個應用的生命周期
  3. 可以被grid中的所有線程訪問,同時還host端還可以通過runtime庫訪問
__shared__

可以和__device__修飾符一同使用

  1. 在block中的共享內存空間中
  2. 擁有block的聲明周期
  3. 只能被block中的線程訪問
__managed__

可以和__device__修飾符一同使用

  1. device和host都可以訪問
  2. 擁有整個應用的生命周期

GPU 編程的步驟

  • 在host上設定輸入數據
  • 在host上分配內存作為輸出 (malloc)
  • 在GPU上分配輸入/輸出內存 (cudaMalloc)
  • 把host上的輸入復制到GPU(cudaMemcpy)
  • 運行GPU kernel
  • 把輸出從GPU復制到host (cudaMemcpy)
device對應的cudaMemcpy/cudaFree
cudaError_t cudaMalloc(void **devPtr, size_t size)
Returns
cudaSuccess, cudaErrorInvalidDevicePointer, cudaErrorInitializationError

cudaError_t cudaFree (void *devPtr)
Returns
cudaSuccess, cudaErrorInvalidDevicePointer, cudaErrorInitializationError

例子:
cudaError_t err = cudaMalloc((void **)&d_A, size);
err = cudaFree(d_A);
host下對應的malloc/free
float *h_A = (float *)malloc(size);
free(h_A);
數據傳輸cudaMemcpy
cudaMemcpy(void *dst, const void *src, size_t count, enum cudaMemcpyKind kind);

enum __device_builtin__ cudaMemcpyKind
{
    cudaMemcpyHostToHost          =   0,      /**< Host   -> Host */
    cudaMemcpyHostToDevice        =   1,      /**< Host   -> Device */
    cudaMemcpyDeviceToHost        =   2,      /**< Device -> Host */
    cudaMemcpyDeviceToDevice      =   3,      /**< Device -> Device */
    cudaMemcpyDefault             =   4       /**< Direction of the transfer is inferred from the pointer values. Requires unified virtual addressing */
};

cudaMemcpy(d_A, h_A, size, cudaMemcpyHostToDevice)
錯誤類型cudaError_t
typedef __device_builtin__ enum cudaError cudaError_t;

enum __device_builtin__ cudaError
{
    cudaSuccess                           =      0,
    cudaErrorMissingConfiguration         =      1,
    cudaErrorMemoryAllocation             =      2,
    ......

};
char* cudaGetErrorName(cudaError_t error)
char* cudaGetErrorString(cudaError_t error)
cudaError_t cudaGetLastError (void)
例子
if (err != cudaSuccess)
{
    fprintf(stderr, "Failed to allocate device vector A (error code %s)!\n", cudaGetErrorString(err));
    exit(EXIT_FAILURE);
}
通用的CUDA Runtime API錯誤處理
#define gpuErrchk(ans) { gpuAssert((ans), __FILE__, __LINE__); }
inline void gpuAssert(cudaError_t code, const char *file, int line, bool abort=true)
{
   if (code != cudaSuccess) 
   {
      fprintf(stderr,"GPUassert: %s %s %d\n", cudaGetErrorString(code), file, line);
      if (abort) exit(code);
   }
}
調用Kernel

<<<blocksPerGrid, threadsPerBlock>>>設定執行次數

int threadsPerBlock = 256;
int blocksPerGrid =(numElements + threadsPerBlock - 1) / threadsPerBlock;
printf("CUDA kernel launch with %d blocks of %d threads\n", blocksPerGrid, threadsPerBlock);
vectorAdd<<<blocksPerGrid, threadsPerBlock>>>(d_A, d_B, d_C, numElements);
err = cudaGetLastError();

__global__ void
vectorAdd(const float *A, const float *B, float *C, int numElements)
{
    int i = blockDim.x * blockIdx.x + threadIdx.x;
    if (i < numElements)
    {
        C[i] = A[i] + B[i];
    }
}
其他

cudaDeviceSynchronize() 阻塞直到device完成之前所有的任務
cudaDeviceReset() 銷毀分配的資源,重置狀態,最后調用
#include "cuda_runtime.h"

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容