Merge pull request #14927 from sneaxiy/fix_cuda_stream_callback_in_cuda10

Fix stream_callback_manager bug in CUDA 10
6 years ago · 36a1d021a4
parent fa78fc60be 2373aeb5e8
commit 36a1d021a4
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/platform/stream_callback_manager.cc
+++ b/paddle/fluid/platform/stream_callback_manager.cc
@ -19,7 +19,7 @@ namespace paddle {
 namespace platform {

 #if CUDA_VERSION >= 10000
-static void CUDART_CB StreamCallbackFunc(void *user_data);
+static void CUDART_CB StreamCallbackFunc(void *user_data)
 #else
 static void CUDART_CB StreamCallbackFunc(cudaStream_t stream,
                                         cudaError_t status, void *user_data)