From 2950bdc1f7dd52ef98ae2ced209813c6bb5f5bdc Mon Sep 17 00:00:00 2001 From: root Date: Mon, 2 Nov 2020 19:33:10 +0000 Subject: [PATCH] [igemm] fix xdlop wrw padding kernel --- ...conv_hip_implicit_gemm_wrw_v4r4_xdlops_padded_gemm.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/solver/conv_hip_implicit_gemm_wrw_v4r4_xdlops_padded_gemm.cpp b/src/solver/conv_hip_implicit_gemm_wrw_v4r4_xdlops_padded_gemm.cpp index 1541e935a6..a20be6d076 100644 --- a/src/solver/conv_hip_implicit_gemm_wrw_v4r4_xdlops_padded_gemm.cpp +++ b/src/solver/conv_hip_implicit_gemm_wrw_v4r4_xdlops_padded_gemm.cpp @@ -42,7 +42,7 @@ namespace solver { PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm::PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm() : PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm:: PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm( - 64, 64, 2, 32, 32, 4, 16, 64, 16, false, false) + 64, 64, 2, 32, 32, 4, 16, 64, 32, false, false) { // GemmMFactor GemmNFactor, GemmKFactor are fixed value at this moment } @@ -126,7 +126,7 @@ void PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm::EuristicInit(const Convol auto get_euristic_config = [&](auto is_valid_func) { if(ctx.IsFp32()) { - tmp = {256, 256, 8, 128, 128, 4, 16, 64, 16, false, true}; + tmp = {256, 256, 8, 128, 128, 4, 16, 64, 32, false, true}; bool all_visited = false; do @@ -157,7 +157,7 @@ void PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm::EuristicInit(const Convol } else if(ctx.IsFp16()) { - tmp = {256, 256, 8, 128, 128, 8, 16, 64, 16, false, true}; + tmp = {256, 256, 8, 128, 128, 8, 16, 64, 32, false, true}; bool all_visited = false; do { @@ -187,7 +187,7 @@ void PerformanceImplicitGemmWrwV4R4Xdlops_Padded_Gemm::EuristicInit(const Convol } else if(ctx.IsBfp16()) { - tmp = {256, 256, 8, 128, 128, 8, 16, 64, 16, false, true}; + tmp = {256, 256, 8, 128, 128, 8, 16, 64, 32, false, true}; bool all_visited = false; do