提交 e6ca1825 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Delete the cudnn handles when we no longer need them.

上级 d6b2e034
...@@ -125,3 +125,7 @@ cudnnHandle_t APPLY_SPECIFIC(_handle); ...@@ -125,3 +125,7 @@ cudnnHandle_t APPLY_SPECIFIC(_handle);
} }
cuda_exit(PARAMS->ctx); cuda_exit(PARAMS->ctx);
} }
#section cleanup_code_struct
cudnnDestroy(APPLY_SPECIFIC(_handle));
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论