メモリの確保・破棄は「cublasAlloc()関数」、「cublasFree()関数」で行う。
■cublasStatus CUBLASAPI cublasAlloc (int n, int elemSize, void **devicePtr)
- 要素がn個の配列のオブジェクトをGPUメモリに作成する。
- elemSize:バイト数
■cublasStatus CUBLASAPI cublasFree (const void *devicePtr)
[例]
float* GpuData = NULL;
... ...
cublasStatus status;
int n = 256*256;
status = cublasAlloc(n, sizeof(GpuData[0]), (void**)&GpuData); //メモリ確保
if(status != CUBLAS_STATUS_SUCCESS){
//---- エラー処理 ----/
}
... ...
status = cublasFree(GpuData); //メモリ破棄
if(status != CUBLAS_STATUS_SUCCESS){
//---- エラー処理 ----/
}
... ...
次 >>
CUBLAS::データ設定