Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 92fdfe33 编写于 作者: X XiaoguangHu 提交者： GitHub
浏览文件
操作

fix bug when compiling with cusparse in CUDA version >=11.4 (#42455)

上级 b0a64800
隐藏空白更改
内联 并排
Showing with 10 addition and 8 deletion
paddle/fluid/platform/dynload/cusparse.cc
浏览文件 @ 92fdfe33
...@@ -24,13 +24,14 @@ namespace dynload { ...@@ -24,13 +24,14 @@ namespace dynload {
CUSPARSE_ROUTINE_EACH(DEFINE_WRAP); CUSPARSE_ROUTINE_EACH(DEFINE_WRAP);
#endif #endif
#ifdef CUBLAS_BLAS_ROUTINE_EACH_R2
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
#endif
#ifdef CUSPARSE_ROUTINE_EACH_11020 #ifdef CUSPARSE_ROUTINE_EACH_11020
CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP); CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP);
#endif #endif
#ifdef CUSPARSE_ROUTINE_EACH_R2
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
#endif
} // namespace dynload } // namespace dynload
} // namespace platform } // namespace platform
} // namespace paddle } // namespace paddle
paddle/phi/backends/dynload/cusparse.cc
浏览文件 @ 92fdfe33
...@@ -26,12 +26,13 @@ void *cusparse_dso_handle; ...@@ -26,12 +26,13 @@ void *cusparse_dso_handle;
CUSPARSE_ROUTINE_EACH(DEFINE_WRAP); CUSPARSE_ROUTINE_EACH(DEFINE_WRAP);
#endif #endif
#ifdef CUBLAS_BLAS_ROUTINE_EACH_R2
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
#endif
#ifdef CUSPARSE_ROUTINE_EACH_11020 #ifdef CUSPARSE_ROUTINE_EACH_11020
CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP); CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP);
#endif #endif
#ifdef CUSPARSE_ROUTINE_EACH_R2
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
#endif
} // namespace dynload } // namespace dynload
} // namespace phi } // namespace phi
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或