未验证 提交 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;
}; \
extern DynLoad__##__name __name
#ifndef _WIN32
#if CUDA_VERSION >= 11020
#define CUSPARSE_ROUTINE_EACH(__macro) \
__macro(cusparseCreate); \
......@@ -57,6 +58,7 @@ extern void *cusparse_dso_handle;
CUSPARSE_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUSPARSE_WRAP);
#endif
#endif
#undef DECLARE_DYNAMIC_LOAD_CUSPARSE_WRAP
} // namespace dynload
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册