提交 672eb0ad authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Remove parameter in call to cublas_shutdown(). There shouldn't be one.

上级 bd7d94df
...@@ -3134,7 +3134,7 @@ CudaNdarray_active_device_name(PyObject* _unused, PyObject* _unused_args) { ...@@ -3134,7 +3134,7 @@ CudaNdarray_active_device_name(PyObject* _unused, PyObject* _unused_args) {
PyObject * PyObject *
CudaNdarray_gpu_shutdown(PyObject* _unused, PyObject* _unused_args) { CudaNdarray_gpu_shutdown(PyObject* _unused, PyObject* _unused_args) {
// Don't handle errors here // Don't handle errors here
cublas_shutdown(handle); cublas_shutdown();
handle = NULL; handle = NULL;
cudaThreadExit(); cudaThreadExit();
g_gpu_context_active = 0; // context has now been closed down g_gpu_context_active = 0; // context has now been closed down
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论