diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-11-15 14:26:44 +0000 |
---|---|---|
committer | SiCong Li <sicong.li@arm.com> | 2019-11-18 14:29:50 +0000 |
commit | 09471fa63e265c4e155e8541ae897b6105293a7d (patch) | |
tree | b302df6a5bb71d16042c4fd49eaf526e99ec1db1 | |
parent | 002cbd34a4b59910bc7ab348f042505fd32de56a (diff) | |
download | ComputeLibrary-09471fa63e265c4e155e8541ae897b6105293a7d.tar.gz |
COMPMID-2853: Fix VGG16 for fp16.
Half the number of granules for fp16 compaired to fp32.
Change-Id: Id9a39d29a5b08295fade8271b93ef80a19a536ce
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2298
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp index 88e060109a..24254eb9e5 100644 --- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp @@ -390,10 +390,10 @@ void Fallback<TypeInput, TypeOutput, OutputStage>::run() // Schedule assembly kernel IScheduler::Hints scheduling_hint = IScheduler::Hints(Window::DimX); - if(_kernel_info.method == arm_gemm::GemmMethod::GEMM_INTERLEAVED) + if(_kernel_info.method == arm_gemm::GemmMethod::GEMM_INTERLEAVED && _d->info()->data_type() == DataType::F32) { - constexpr int granule_threshold = 200; - scheduling_hint = IScheduler::Hints(Window::DimX, IScheduler::StrategyHint::DYNAMIC, granule_threshold); + const int granule_threshold = 200; + scheduling_hint = IScheduler::Hints(Window::DimX, IScheduler::StrategyHint::DYNAMIC, granule_threshold); } NEScheduler::get().schedule(_optimised_kernel.get(), scheduling_hint); } |