Free CUDA buffers

This commit is contained in:
Tom Deakin 2015-07-31 13:59:34 +01:00
parent 07fb03c134
commit 70ccd22919

View File

@ -332,6 +332,14 @@ int main(int argc, char *argv[])
free(h_b);
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)