[CUDA] Free the sum device buffer
This commit is contained in:
parent
d75b0004de
commit
b9c514fd9b
@ -77,6 +77,8 @@ CUDAStream<T>::~CUDAStream()
|
|||||||
check_error();
|
check_error();
|
||||||
cudaFree(d_c);
|
cudaFree(d_c);
|
||||||
check_error();
|
check_error();
|
||||||
|
cudaFree(d_sum);
|
||||||
|
check_error();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user