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

fix_crf_doc
zhouwei25 6 years ago committed by liuwei1031
parent 43a82d83fa
commit 84c728013c

@ -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

Loading…
Cancel
Save