aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGian Marco Iodice <gianmarco.iodice@arm.com>2019-08-13 11:43:38 +0100
committerGian Marco Iodice <gianmarco.iodice@arm.com>2019-08-13 12:07:19 +0000
commit670c04ac4e99ee4b9eeb231e9d980c927bddfaa0 (patch)
treebcb6081e100f3755a803e8642a9cbd5127f7c6e6
parent93c70b8d7c69ddcee58162c2618aa0e898b4bd1e (diff)
downloadComputeLibrary-670c04ac4e99ee4b9eeb231e9d980c927bddfaa0.tar.gz
COMPMID-2561: Fix GEMM heuristic
Change-Id: Ideb8a4e56d85de7c3cd08e31ac656658ca88119b Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com> Reviewed-on: https://review.mlplatform.org/c/1724 Reviewed-by: Pablo Marquez <pablo.tello@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r--arm_compute/core/CL/gemm/native/CLGEMMNativeKernelConfiguration.h10
-rw-r--r--arm_compute/core/CL/gemm/reshaped/CLGEMMReshapedKernelConfiguration.h10
-rw-r--r--arm_compute/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfiguration.h10
3 files changed, 9 insertions, 21 deletions
diff --git a/arm_compute/core/CL/gemm/native/CLGEMMNativeKernelConfiguration.h b/arm_compute/core/CL/gemm/native/CLGEMMNativeKernelConfiguration.h
index 7d0e7c97d4..763c27b0fb 100644
--- a/arm_compute/core/CL/gemm/native/CLGEMMNativeKernelConfiguration.h
+++ b/arm_compute/core/CL/gemm/native/CLGEMMNativeKernelConfiguration.h
@@ -45,13 +45,9 @@ public:
*/
static std::unique_ptr<ICLGEMMKernelConfiguration> create(GPUTarget arch)
{
- switch(get_arch_from_target(arch))
- {
- case GPUTarget::BIFROST:
- return support::cpp14::make_unique<CLGEMMNativeKernelConfigurationBifrost>(arch);
- default:
- return nullptr;
- }
+ // Note: At the moment we only support Bifrost architecture. However, we should have a dedicated path for each GPU architecture
+ // using get_arch_from_target(arch)
+ return support::cpp14::make_unique<CLGEMMNativeKernelConfigurationBifrost>(arch);
}
};
} // namespace cl_gemm
diff --git a/arm_compute/core/CL/gemm/reshaped/CLGEMMReshapedKernelConfiguration.h b/arm_compute/core/CL/gemm/reshaped/CLGEMMReshapedKernelConfiguration.h
index 105a58a6f8..f451a18d74 100644
--- a/arm_compute/core/CL/gemm/reshaped/CLGEMMReshapedKernelConfiguration.h
+++ b/arm_compute/core/CL/gemm/reshaped/CLGEMMReshapedKernelConfiguration.h
@@ -45,13 +45,9 @@ public:
*/
static std::unique_ptr<ICLGEMMKernelConfiguration> create(GPUTarget arch)
{
- switch(get_arch_from_target(arch))
- {
- case GPUTarget::BIFROST:
- return support::cpp14::make_unique<CLGEMMReshapedKernelConfigurationBifrost>(arch);
- default:
- return nullptr;
- }
+ // Note: At the moment we only support Bifrost architecture. However, we should have a dedicated path for each GPU architecture
+ // using get_arch_from_target(arch)
+ return support::cpp14::make_unique<CLGEMMReshapedKernelConfigurationBifrost>(arch);
}
};
} // namespace cl_gemm
diff --git a/arm_compute/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfiguration.h b/arm_compute/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfiguration.h
index b9bf150c4f..5d9eac4d02 100644
--- a/arm_compute/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfiguration.h
+++ b/arm_compute/core/CL/gemm/reshaped_only_rhs/CLGEMMReshapedOnlyRHSKernelConfiguration.h
@@ -45,13 +45,9 @@ public:
*/
static std::unique_ptr<ICLGEMMKernelConfiguration> create(GPUTarget arch)
{
- switch(get_arch_from_target(arch))
- {
- case GPUTarget::BIFROST:
- return support::cpp14::make_unique<CLGEMMReshapedOnlyRHSKernelConfigurationBifrost>(arch);
- default:
- return nullptr;
- }
+ // Note: At the moment we only support Bifrost architecture. However, we should have a dedicated path for each GPU architecture
+ // using get_arch_from_target(arch)
+ return support::cpp14::make_unique<CLGEMMReshapedOnlyRHSKernelConfigurationBifrost>(arch);
}
};
} // namespace cl_gemm