Skip to content

Commit b57c132

Browse files
fix bug when compiling with cusparse in CUDA version >=11.4 (#42456)
1 parent eae41b7 commit b57c132

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

paddle/fluid/platform/dynload/cusparse.cc

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,14 @@ namespace dynload {
2424
CUSPARSE_ROUTINE_EACH(DEFINE_WRAP);
2525
#endif
2626

27-
#ifdef CUBLAS_BLAS_ROUTINE_EACH_R2
28-
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
29-
#endif
30-
3127
#ifdef CUSPARSE_ROUTINE_EACH_11020
3228
CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP);
3329
#endif
30+
31+
#ifdef CUSPARSE_ROUTINE_EACH_R2
32+
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
33+
#endif
34+
3435
} // namespace dynload
3536
} // namespace platform
3637
} // namespace paddle

paddle/phi/backends/dynload/cusparse.cc

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,13 @@ void *cusparse_dso_handle;
2626
CUSPARSE_ROUTINE_EACH(DEFINE_WRAP);
2727
#endif
2828

29-
#ifdef CUBLAS_BLAS_ROUTINE_EACH_R2
30-
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
31-
#endif
32-
3329
#ifdef CUSPARSE_ROUTINE_EACH_11020
3430
CUSPARSE_ROUTINE_EACH_11020(DEFINE_WRAP);
3531
#endif
32+
33+
#ifdef CUSPARSE_ROUTINE_EACH_R2
34+
CUSPARSE_ROUTINE_EACH_R2(DEFINE_WRAP);
35+
#endif
36+
3637
} // namespace dynload
3738
} // namespace phi

0 commit comments

Comments
 (0)