diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-08-13 11:43:38 +0100 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-08-13 12:07:19 +0000 |
commit | 670c04ac4e99ee4b9eeb231e9d980c927bddfaa0 (patch) | |
tree | bcb6081e100f3755a803e8642a9cbd5127f7c6e6 | |
parent | 93c70b8d7c69ddcee58162c2618aa0e898b4bd1e (diff) | |
download | ComputeLibrary-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>
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 |