From 670c04ac4e99ee4b9eeb231e9d980c927bddfaa0 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Tue, 13 Aug 2019 11:43:38 +0100 Subject: COMPMID-2561: Fix GEMM heuristic Change-Id: Ideb8a4e56d85de7c3cd08e31ac656658ca88119b Signed-off-by: Gian Marco Iodice Reviewed-on: https://review.mlplatform.org/c/1724 Reviewed-by: Pablo Marquez Reviewed-by: Michele Di Giorgio Tested-by: Arm Jenkins --- .../core/CL/gemm/native/CLGEMMNativeKernelConfiguration.h | 10 +++------- .../core/CL/gemm/reshaped/CLGEMMReshapedKernelConfiguration.h | 10 +++------- .../CLGEMMReshapedOnlyRHSKernelConfiguration.h | 10 +++------- 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 create(GPUTarget arch) { - switch(get_arch_from_target(arch)) - { - case GPUTarget::BIFROST: - return support::cpp14::make_unique(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(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 create(GPUTarget arch) { - switch(get_arch_from_target(arch)) - { - case GPUTarget::BIFROST: - return support::cpp14::make_unique(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(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 create(GPUTarget arch) { - switch(get_arch_from_target(arch)) - { - case GPUTarget::BIFROST: - return support::cpp14::make_unique(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(arch); } }; } // namespace cl_gemm -- cgit v1.2.1