aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2019-11-15 14:26:44 +0000
committerGeorgios Pinitas <georgios.pinitas@arm.com>2019-11-18 14:01:24 +0000
commit6011f2444eacd8e549c74bdbb10e1c3cd2b29e77 (patch)
treeb302df6a5bb71d16042c4fd49eaf526e99ec1db1
parent46023ed57210a0f72e6df457af8a8cd966d6565b (diff)
downloadComputeLibrary-6011f2444eacd8e549c74bdbb10e1c3cd2b29e77.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.cpp6
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);
}