diff options
| author | Young Jin Kim <youki@microsoft.com> | 2020-01-18 21:12:49 +0300 |
|---|---|---|
| committer | Young Jin Kim <youki@microsoft.com> | 2020-01-18 21:12:49 +0300 |
| commit | 1a5d2d5efcaf7336d7f652b5e8eb25f4205a628d (patch) | |
| tree | c176b197d651ce9a364c368295815332747c4eee | |
| parent | 8236d1bb9ba4ae6e37e27b0f26f038ceeff48753 (diff) | |
Add missed AVX2 fp16 callsyouki/fix-avx2-fp16
| -rw-r--r-- | src/FbgemmFP16UKernelsIntrinsicAvx2.cc | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/FbgemmFP16UKernelsIntrinsicAvx2.cc b/src/FbgemmFP16UKernelsIntrinsicAvx2.cc index f41419e..e5f483b 100644 --- a/src/FbgemmFP16UKernelsIntrinsicAvx2.cc +++ b/src/FbgemmFP16UKernelsIntrinsicAvx2.cc @@ -92,5 +92,23 @@ void gemmkernel_Avx2_fp16_fA0fB0fC0( } } +void NOINLINE gemmkernel_1x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp) { + gemmkernel_Avx2_fp16_fA0fB0fC0(gp, 1); +} +void NOINLINE gemmkernel_2x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp) { + gemmkernel_Avx2_fp16_fA0fB0fC0(gp, 2); +} +void NOINLINE gemmkernel_3x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp) { + gemmkernel_Avx2_fp16_fA0fB0fC0(gp, 3); +} +void NOINLINE gemmkernel_4x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp) { + gemmkernel_Avx2_fp16_fA0fB0fC0(gp, 4); +} +void NOINLINE gemmkernel_5x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp) { + gemmkernel_Avx2_fp16_fA0fB0fC0(gp, 5); +} +void NOINLINE gemmkernel_6x2_Avx2_fp16_fA0fB0fC0(GemmParamsFP16* gp) { + gemmkernel_Avx2_fp16_fA0fB0fC0(gp, 6); +} } // namespace fbgemm #endif // _MSC_VER |