diff options
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/CL/CLHelpers.h | 7 | ||||
-rw-r--r-- | arm_compute/core/GPUTarget.h | 4 |
2 files changed, 10 insertions, 1 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 */ |