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

b57c132a · XiaoguangHu · GitHub · eae41b7d · b57c132a · b57c132a
隐藏空白更改
内联并排

Showing with 10 addition and 8 deletion

paddle/fluid/platform/dynload/cusparse.cc paddle/fluid/platform/dynload/cusparse.cc +5 -4

paddle/phi/backends/dynload/cusparse.cc paddle/phi/backends/dynload/cusparse.cc +5 -4

未找到文件。
--- a/paddle/fluid/platform/dynload/cusparse.cc
+++ b/paddle/fluid/platform/dynload/cusparse.cc
@@ -24,13 +24,14 @@ namespace dynload {
 CUSPARSE_ROUTINE_EACH(DEFINE_WRAP);
 #endif

-#ifdef CUBLAS_BLAS_ROUTINE_EACH_R2
-CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
-#endif
-
 #ifdef CUSPARSE_ROUTINE_EACH_11020
 CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP);
 #endif
+
+#ifdef CUSPARSE_ROUTINE_EACH_R2
+CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
+#endif
+
 }  // namespace dynload
 }  // namespace platform
 }  // namespace paddle
--- a/paddle/phi/backends/dynload/cusparse.cc
+++ b/paddle/phi/backends/dynload/cusparse.cc
@@ -26,12 +26,13 @@ void *cusparse_dso_handle;
 CUSPARSE_ROUTINE_EACH(DEFINE_WRAP);
 #endif

-#ifdef CUBLAS_BLAS_ROUTINE_EACH_R2
-CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
-#endif
-
 #ifdef CUSPARSE_ROUTINE_EACH_11020
 CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP);
 #endif
+
+#ifdef CUSPARSE_ROUTINE_EACH_R2
+CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
+#endif
+
 }  // namespace dynload
 }  // namespace phi