1 | /******************************************************************************* |
2 | * Copyright 2017-2021 Intel Corporation |
3 | * |
4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
5 | * you may not use this file except in compliance with the License. |
6 | * You may obtain a copy of the License at |
7 | * |
8 | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | * |
10 | * Unless required by applicable law or agreed to in writing, software |
11 | * distributed under the License is distributed on an "AS IS" BASIS, |
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
13 | * See the License for the specific language governing permissions and |
14 | * limitations under the License. |
15 | *******************************************************************************/ |
16 | |
17 | #ifndef CPU_X64_GEMM_F32_JIT_AVX512_COMMON_GEMM_F32_HPP |
18 | #define CPU_X64_GEMM_F32_JIT_AVX512_COMMON_GEMM_F32_HPP |
19 | |
20 | #include "oneapi/dnnl/dnnl_types.h" |
21 | |
22 | #include "cpu/gemm/f32/gemm_utils_f32.hpp" |
23 | |
24 | namespace dnnl { |
25 | namespace impl { |
26 | namespace cpu { |
27 | namespace x64 { |
28 | |
29 | dnnl_status_t jit_avx512_common_gemm_f32(int nthrs, const char *transa, |
30 | const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, |
31 | const float *alpha, const float *A, const dim_t *lda, const float *B, |
32 | const dim_t *ldb, const float *beta, float *C, const dim_t *ldc, |
33 | const float *bias = nullptr); |
34 | |
35 | namespace avx512_common_gemm_f32 { |
36 | |
37 | dnnl_status_t sgemm_nocopy_driver(const char *transa, const char *transb, |
38 | dim_t m, dim_t n, dim_t k, const float *alpha, const float *a, |
39 | dim_t lda, const float *b, dim_t ldb, const float *beta, float *c, |
40 | dim_t ldc, const float *bias); |
41 | } |
42 | |
43 | } // namespace x64 |
44 | } // namespace cpu |
45 | } // namespace impl |
46 | } // namespace dnnl |
47 | |
48 | #endif // CPU_X64_GEMM_F32_JIT_AVX512_COMMON_GEMM_F32_HPP |
49 | |