提交 4db43c6c 编写于 作者: Y Yu Yang

Naive implement cblas

上级 60d6348e
...@@ -24,17 +24,23 @@ struct CBlas; ...@@ -24,17 +24,23 @@ struct CBlas;
template <> template <>
struct CBlas<float> { struct CBlas<float> {
static constexpr auto GEMM = cblas_sgemm; template <typename... ARGS>
static void GEMM(ARGS... args) {
cblas_sgemm(args...);
}
}; };
template <> template <>
struct CBlas<double> { struct CBlas<double> {
static constexpr auto GEMM = cblas_dgemm; template <typename... ARGS>
static void GEMM(ARGS... args) {
cblas_dgemm(args...);
}
}; };
template <> template <>
struct CBlas<platform::float16> { struct CBlas<platform::float16> {
void GEMM(...) { PADDLE_THROW("float16 GEMM not supported on CPU"); } static void GEMM(...) { PADDLE_THROW("float16 GEMM not supported on CPU"); }
}; };
template <> template <>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册