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

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

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