diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-05-05 15:10:21 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-05-05 16:26:36 +0000 |
| commit | 939586e8fd05923cab63b7b79bf5040cca5b8a7b (patch) | |
| tree | 6a04498284521bf2fbc0a5c1c615d62489bf1954 | |
| parent | 5c2df4e8aa3655800e764a29ad25cdbc560558be (diff) | |
| download | ComputeLibrary-939586e8fd05923cab63b7b79bf5040cca5b8a7b.tar.gz | |
COMPMID-3428: Fixed Mali-G77 heuristic for GEMMReshapedOnlyRHS - f16
Change-Id: Ie1f021235edf3d60171a252d519c53df2d7fb0c6
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3146
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfigurationValhall.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfigurationValhall.cpp b/src/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfigurationValhall.cpp index 783d0fe91b..5526616001 100644 --- a/src/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfigurationValhall.cpp +++ b/src/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfigurationValhall.cpp @@ -112,7 +112,7 @@ std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> CLGEMMReshapedOnlyRHSKernelConfi const int h0 = std::max(std::min(static_cast<int>(n / 4), static_cast<int>(256)), static_cast<int>(1)); return configure_lhs_rhs_info(m, n, 1, 4, 4, 1, h0, false, true, false, true); } - else + else if (m < 128) { const int h0 = std::max(std::min(static_cast<int>(n / 4), static_cast<int>(256)), static_cast<int>(1)); if(k > 512) @@ -124,6 +124,11 @@ std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> CLGEMMReshapedOnlyRHSKernelConfi return configure_lhs_rhs_info(m, n, 2, 4, 8, 1, h0, false, true, false, false); } } + else + { + const int h0 = std::max(std::min(static_cast<int>(n / 4), static_cast<int>(256)), static_cast<int>(1)); + return configure_lhs_rhs_info(m, n, 4, 4, 4, 1, h0, false, true, false, false); + } } std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> CLGEMMReshapedOnlyRHSKernelConfigurationValhall::configure_G77_u8(unsigned int m, unsigned int n, unsigned int k, unsigned int b) |