commit 09471fa63e265c4e155e8541ae897b6105293a7d
author Georgios Pinitas <georgios.pinitas@arm.com> Fri Nov 15 14:26:44 2019 +0000
committer SiCong Li <sicong.li@arm.com> Mon Nov 18 14:29:50 2019 +0000
tree b302df6a5bb71d16042c4fd49eaf526e99ec1db1
parent 002cbd34a4b59910bc7ab348f042505fd32de56a

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>

src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp

diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp
index 88e0601..24254eb 100644
--- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp
+++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp
@@ -390,10 +390,10 @@
 
     // 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);
 }