Skip to content

GitLab

"doc/git@developer.sourcefind.cn:OpenDAS/ktransformers.git" did not exist on "bbcc5481bbe40c7d92601671da1b3190abf33986"
Commit 1bde1726 authored by Jiezhong Qiu's avatar Jiezhong Qiu
Browse files

cudafree malloced memory

parent 97a2b1be
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
pytorch/cuda/moe_cuda_kernel.cu
View file @ 1bde1726
...@@ -85,7 +85,10 @@ void moe_cuda_forward_impl( ...@@ -85,7 +85,10 @@ void moe_cuda_forward_impl(
Carray, 1, Carray, 1,
batch_size)); batch_size));
checkCudaErrors(cudaStreamSynchronize(*(h->streams))); checkCudaErrors(cudaStreamSynchronize(*(h->streams)));
checkCudaErrors(cudaFree(Aarray));
checkCudaErrors(cudaFree(Barray));
checkCudaErrors(cudaFree(Carray));
} }
template <typename scalar_t> template <typename scalar_t>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment