提交 84c72801 编写于 作者: Z zhouwei25 提交者: liuwei1031

fix the compilation issue on windows caused by mkl_CSRMM (#19533)

上级 43a82d83
......@@ -128,11 +128,12 @@ struct CBlas<float> {
static void VMERF(ARGS... args) {
platform::dynload::vmsErf(args...);
}
#if !defined(_WIN32)
template <typename... ARGS>
static void CSRMM(ARGS... args) {
platform::dynload::mkl_scsrmm(args...);
}
#endif
};
template <>
......@@ -238,11 +239,12 @@ struct CBlas<double> {
static void VMERF(ARGS... args) {
platform::dynload::vmdErf(args...);
}
#if !defined(_WIN32)
template <typename... ARGS>
static void CSRMM(ARGS... args) {
platform::dynload::mkl_dcsrmm(args...);
}
#endif
};
#else
......
......@@ -88,12 +88,15 @@ extern void* mklml_dso_handle;
__macro(vdInv); \
__macro(vmsErf); \
__macro(vmdErf); \
__macro(mkl_scsrmm); \
__macro(mkl_dcsrmm); \
__macro(MKL_Set_Num_Threads)
MKLML_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_MKLML_WRAP);
#if !defined(_WIN32)
DYNAMIC_LOAD_MKLML_WRAP(mkl_scsrmm);
DYNAMIC_LOAD_MKLML_WRAP(mkl_dcsrmm);
#endif
#undef DYNAMIC_LOAD_MKLML_WRAP
} // namespace dynload
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册