diff options
| author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2022-10-21 11:15:54 +0100 |
|---|---|---|
| committer | Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2022-11-01 10:58:52 +0000 |
| commit | 4b5f6efef15efd79727a58c520c92c9e7a084256 (patch) | |
| tree | b8acdadcd397b992f2e7148a2f4da088cb7ae7d5 /src/cpu/operators/internal/CpuGemmAssemblyDispatch.h | |
| parent | f44bbc5c697de841dce97c0f2fa39bae391a8174 (diff) | |
| download | ComputeLibrary-4b5f6efef15efd79727a58c520c92c9e7a084256.tar.gz | |
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>
Diffstat (limited to 'src/cpu/operators/internal/CpuGemmAssemblyDispatch.h')
| -rw-r--r-- | src/cpu/operators/internal/CpuGemmAssemblyDispatch.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h index 691eeff8d2..0c51c92359 100644 --- a/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h +++ b/src/cpu/operators/internal/CpuGemmAssemblyDispatch.h @@ -54,6 +54,7 @@ struct AsmGemmInfo 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 */ |