commit | 4b5f6efef15efd79727a58c520c92c9e7a084256 | [log] [tgz] |
---|---|---|
author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | Fri Oct 21 11:15:54 2022 +0100 |
committer | Suhail Munshi <MohammedSuhail.Munshi@arm.com> | Tue Nov 01 10:58:52 2022 +0000 |
tree | b8acdadcd397b992f2e7148a2f4da088cb7ae7d5 | |
parent | f44bbc5c697de841dce97c0f2fa39bae391a8174 [diff] [blame] |
Add check for Batch Matmul in GemmAssemblyDispatch Relates to : COMPMID-5507 Change-Id: Ia2c4ea153ac2524ffa6b2a9c10f3a0318a8a67a1 Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8509 Benchmark: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: SiCong Li <sicong.li@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h index 691eeff..0c51c92 100644 --- a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h +++ b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h
@@ -54,6 +54,7 @@ bool fast_mode{ false }; bool fixed_format{ false }; arm_compute::WeightFormat weight_format{ arm_compute::WeightFormat::UNSPECIFIED }; + bool reshape_b_only_on_first_run{ true }; }; /** Assembly kernel glue */