未验证 提交 0e19aeb9 编写于 作者: L Liu-xiandong 提交者: GitHub

[cherry-pick] fix cusparse compile bug in windows CUDA11.2, test=release/2.2 (#36015)

解决Windows中CUDA11.2编译出错的问题。
cherry-pick #35941
上级 063fca8e
...@@ -41,6 +41,7 @@ extern void *cusparse_dso_handle; ...@@ -41,6 +41,7 @@ extern void *cusparse_dso_handle;
}; \ }; \
extern DynLoad__##__name __name extern DynLoad__##__name __name
#ifndef _WIN32
#if CUDA_VERSION >= 11020 #if CUDA_VERSION >= 11020
#define CUSPARSE_ROUTINE_EACH(__macro) \ #define CUSPARSE_ROUTINE_EACH(__macro) \
__macro(cusparseCreate); \ __macro(cusparseCreate); \
...@@ -57,6 +58,7 @@ extern void *cusparse_dso_handle; ...@@ -57,6 +58,7 @@ extern void *cusparse_dso_handle;
CUSPARSE_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUSPARSE_WRAP); CUSPARSE_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUSPARSE_WRAP);
#endif #endif
#endif
#undef DECLARE_DYNAMIC_LOAD_CUSPARSE_WRAP #undef DECLARE_DYNAMIC_LOAD_CUSPARSE_WRAP
} // namespace dynload } // namespace dynload
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册