1 | /* |
---|---|
2 | * Copyright (c) Meta Platforms, Inc. and affiliates. |
3 | * All rights reserved. |
4 | * This source code is licensed under the BSD-style license found in the |
5 | * LICENSE file in the root directory of this source tree. |
6 | */ |
7 | #pragma once |
8 | #include <cstdint> |
9 | #include "fbgemm/FbgemmBuild.h" |
10 | #include "fbgemm/FbgemmFPCommon.h" |
11 | #include "fbgemm/Types.h" |
12 | |
13 | namespace fbgemm { |
14 | |
15 | using GemmParamsFP16 = GemmParams<float16>; |
16 | |
17 | void NOINLINE gemmkernel_1x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp); |
18 | void NOINLINE gemmkernel_2x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp); |
19 | void NOINLINE gemmkernel_3x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp); |
20 | void NOINLINE gemmkernel_4x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp); |
21 | void NOINLINE gemmkernel_5x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp); |
22 | void NOINLINE gemmkernel_6x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp); |
23 | |
24 | } // namespace fbgemm |
25 |