未验证 提交 36a1d021 编写于 作者: Z Zeng Jinle 提交者: GitHub

Merge pull request #14927 from sneaxiy/fix_cuda_stream_callback_in_cuda10

Fix stream_callback_manager bug in CUDA 10
...@@ -19,7 +19,7 @@ namespace paddle { ...@@ -19,7 +19,7 @@ namespace paddle {
namespace platform { namespace platform {
#if CUDA_VERSION >= 10000 #if CUDA_VERSION >= 10000
static void CUDART_CB StreamCallbackFunc(void *user_data); static void CUDART_CB StreamCallbackFunc(void *user_data)
#else #else
static void CUDART_CB StreamCallbackFunc(cudaStream_t stream, static void CUDART_CB StreamCallbackFunc(cudaStream_t stream,
cudaError_t status, void *user_data) cudaError_t status, void *user_data)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册