Free CUDA buffers
This commit is contained in:
parent
07fb03c134
commit
70ccd22919
@ -332,6 +332,14 @@ int main(int argc, char *argv[])
|
|||||||
free(h_b);
|
free(h_b);
|
||||||
free(h_c);
|
free(h_c);
|
||||||
|
|
||||||
|
// Free cuda buffers
|
||||||
|
cudaFree(d_a);
|
||||||
|
check_cuda_error();
|
||||||
|
cudaFree(d_b);
|
||||||
|
check_cuda_error();
|
||||||
|
cudaFree(d_c);
|
||||||
|
check_cuda_error();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string getDeviceName(int device)
|
std::string getDeviceName(int device)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user