diff options
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 7 | ||||
| -rw-r--r-- | arm_compute/core/GPUTarget.h | 4 | ||||
| -rw-r--r-- | arm_compute/runtime/CL/CLTypes.h | 6 |
3 files changed, 14 insertions, 3 deletions
diff --git a/arm_compute/core/CL/CLHelpers.h b/arm_compute/core/CL/CLHelpers.h index 94ec5d781b..edbc705c6f 100644 --- a/arm_compute/core/CL/CLHelpers.h +++ b/arm_compute/core/CL/CLHelpers.h @@ -260,5 +260,12 @@ bool export_weights_to_cl_image(const ITensorInfo *tensor); */ void set_unroll_with_pragma(CLBuildOptions &built_opts, std::initializer_list<int> values); +/** Helper function to check whether the cl_arm_matrix_multiply extension is supported + * + * @param[in] device A CL device + * + * @return True if the extension is supported + */ +bool arm_matrix_multiply_supported(const cl::Device &device); } // namespace arm_compute #endif /* ARM_COMPUTE_CLHELPERS_H */ diff --git a/arm_compute/core/GPUTarget.h b/arm_compute/core/GPUTarget.h index 6a8577ac4d..7e2cfe1b6b 100644 --- a/arm_compute/core/GPUTarget.h +++ b/arm_compute/core/GPUTarget.h @@ -51,9 +51,11 @@ enum class GPUTarget G31 = 0x242, G76 = 0x250, G77 = 0x310, + G57 = 0x311, G78 = 0x320, G710 = 0x330, - G57 = 0x340, + G715 = 0x340, + G615 = 0x341 }; /** Enable bitwise operations on GPUTarget enumerations */ diff --git a/arm_compute/runtime/CL/CLTypes.h b/arm_compute/runtime/CL/CLTypes.h index bba25c6d64..d298ecd614 100644 --- a/arm_compute/runtime/CL/CLTypes.h +++ b/arm_compute/runtime/CL/CLTypes.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021 Arm Limited. + * Copyright (c) 2020-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -35,7 +35,9 @@ enum class CLGEMMKernelType /** Reshaped GEMM kernel where both lhs and rhs matrices are reshaped. Configurable reshape and block size */ RESHAPED, /** Reshaped GEMM kernel where only the rhs matrix is reshaped. Configurable reshape and block size */ - RESHAPED_ONLY_RHS + RESHAPED_ONLY_RHS, + /** Reshaped GEMM kernel where only the rhs matrix is reshaped. Using MMUL with configurable block size. */ + RESHAPED_ONLY_RHS_MMUL }; /** OpenCL GEMM kernel selection parameters. These information are retrieved to select the GEMM kernel on OpenCL */ |