Merge pull request #14927 from sneaxiy/fix_cuda_stream_callback_in_cuda10

Fix stream_callback_manager bug in CUDA 10

Merge pull request #14927 from sneaxiy/fix_cuda_stream_callback_in_cuda10
Fix stream_callback_manager bug in CUDA 10
36a1d021 · Zeng Jinle · GitHub · fa78fc60 · 2373aeb5 · 36a1d021
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/platform/stream_callback_manager.cc paddle/fluid/platform/stream_callback_manager.cc +1 -1

未找到文件。
--- 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)