Skip to content

GitLab

Commit a4b52461 authored by Chao Liu's avatar Chao Liu
Browse files

adding implicit GEMM v4r2

parent e87aa851
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 3 deletions
driver/include/device_convolution_implicit_gemm_v4r2_nchw_kcyx_nkhw.hpp
View file @ a4b52461
...@@ -55,13 +55,13 @@ void device_convolution_implicit_gemm_v4r2_nchw_kcyx_nkhw(InDesc, ...@@ -55,13 +55,13 @@ void device_convolution_implicit_gemm_v4r2_nchw_kcyx_nkhw(InDesc,
#if 1 #if 1
// 1x1 filter, 8x8 image // 1x1 filter, 8x8 image
constexpr index_t N1 = 2; constexpr index_t N0 = 1;
constexpr index_t N2 = 1; constexpr index_t N2 = 1;
constexpr index_t Ho1 = 8; constexpr index_t Ho0 = 1;
constexpr index_t Ho2 = 1; constexpr index_t Ho2 = 1;
constexpr index_t Wo1 = 1; constexpr index_t Wo0 = 2;
constexpr index_t Wo2 = 4; constexpr index_t Wo2 = 4;
constexpr index_t BlockSize = 256; constexpr index_t BlockSize = 256;
...@@ -105,6 +105,10 @@ void device_convolution_implicit_gemm_v4r2_nchw_kcyx_nkhw(InDesc, ...@@ -105,6 +105,10 @@ void device_convolution_implicit_gemm_v4r2_nchw_kcyx_nkhw(InDesc,
constexpr index_t WeiBlockCopyDstDataPerWrite_K = 1; constexpr index_t WeiBlockCopyDstDataPerWrite_K = 1;
#endif #endif
constexpr index_t N1 = N / (N0 * N2);
constexpr index_t Ho1 = Ho / (Ho0 * Ho2);
constexpr index_t Wo1 = Wo / (Wo0 * Wo2);
constexpr index_t B = N1 * Ho1 * Wo1; constexpr index_t B = N1 * Ho1 * Wo1;
constexpr index_t GridSize = constexpr index_t GridSize =
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment