24.02.1
|
Data Structures | |
class | ClMatMulNativeConfigArray |
Basic container for the OpenCL MatMul Native configuration functions. More... | |
class | ClMatMulNativeDefaultConfigValhall |
Valhall based OpenCL matmul configuration. More... | |
class | ClMatMulNativeDefaultVariantValhall |
Valhall based OpenCL matmul configuration. More... | |
class | ClMatMulNativeKernelConfigurationFactory |
ClMatMul configuration factory class. More... | |
class | ClMatMulNativeKernelVariantFactory |
ClMatMul variant factory class. More... | |
class | ClMatMulNativeVariantArray |
Basic container for the OpenCL MatMul Native variant functions. More... | |
class | IClMatMulNativeKernelConfig |
Basic interface for the matmul native kernel configuration This is the base class that chooses architecture specific kernel configurations. More... | |
class | IClMatMulNativeKernelVariant |
Basic interface for the matmul native kernel variant This is the base class that chooses architecture specific kernel variants. More... | |
Typedefs | |
using | MatMulNativeConfigsMatrix = std::vector< std::vector< int32_t > > |
Enumerations | |
enum | MatMulKernelType { NATIVE_FP, NATIVE_QUANTIZED, NATIVE_MMUL_FP, NATIVE_MMUL_QUANTIZED } |
Functions | |
MatMulKernelInfo | select_info (const MatMulKernelInfo &info0, const MatMulKernelInfo &info1, unsigned int m, unsigned int n, unsigned int k, unsigned int b, DataType data_type, bool rhs_lock_padding) |
This function accepts two MatMulKernelInfo objects where only the first can be with cl_image2d support enabled. More... | |
MatMulKernelInfo | find_info (const MatMulNativeConfigsMatrix &configs, bool adj_lhs, bool adj_rhs, unsigned int m, unsigned int n, unsigned int k, unsigned int b) |
Find the preferred configurations for the MatMul Native kernel using the MatMulNativeConfigsMatrix provided by the user. More... | |
using MatMulNativeConfigsMatrix = std::vector<std::vector<int32_t> > |
Definition at line 36 of file ClMatMulNativeHelpers.h.
|
strong |
Definition at line 41 of file IClMatMulNativeKernelVariant.h.
MatMulKernelInfo find_info | ( | const MatMulNativeConfigsMatrix & | configs, |
bool | adj_lhs, | ||
bool | adj_rhs, | ||
unsigned int | m, | ||
unsigned int | n, | ||
unsigned int | k, | ||
unsigned int | b | ||
) |
Find the preferred configurations for the MatMul Native kernel using the MatMulNativeConfigsMatrix provided by the user.
[in] | configs | List of best configurations for a limited number of MatMul shapes |
[in] | adj_lhs | Adjoint LHS flag value |
[in] | adj_rhs | Adjoint RHS flag value |
[in] | m | Number of rows (M) of the LHS matrix |
[in] | n | Number of columns (N) in the RHS matrix not reshaped |
[in] | k | Number of rows (K) in the RHS matrix not reshaped |
[in] | b | Batch size |
Definition at line 81 of file ClMatMulNativeHelpers.cpp.
References MatMulKernelInfo::adj_lhs, MatMulKernelInfo::adj_rhs, ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_MSG, ARM_COMPUTE_UNUSED, arm_compute::test::validation::b, MatMulKernelInfo::export_rhs_to_cl_image, MatMulKernelInfo::k0, MatMulKernelInfo::m0, MatMulKernelInfo::n0, and arm_compute::utils::cast::U.
MatMulKernelInfo select_info | ( | const MatMulKernelInfo & | info0, |
const MatMulKernelInfo & | info1, | ||
unsigned int | m, | ||
unsigned int | n, | ||
unsigned int | k, | ||
unsigned int | b, | ||
DataType | data_type, | ||
bool | rhs_lock_padding | ||
) |
This function accepts two MatMulKernelInfo objects where only the first can be with cl_image2d support enabled.
The aim of this function is to check whether the first MatMulKernelInfo object is valid. If not, the function will return the second MatMulKernelInfo object. Otherwise, the first one.
[in] | info0 | MatMulKernelInfo with cl_image2d support |
[in] | info1 | MatMulKernelInfo to fall-back if cl_image2d cannot be used |
[in] | m | Number of rows (M) of the LHS matrix |
[in] | n | Number of columns (N) in the RHS matrix not reshaped |
[in] | k | Number of rows (K) in the RHS matrix not reshaped |
[in] | b | Batch size |
[in] | data_type | Data type |
[in] | rhs_lock_padding | Flag used to know whether the RHS paddings are locked |
Definition at line 39 of file ClMatMulNativeHelpers.cpp.
References MatMulKernelInfo::adj_lhs, MatMulKernelInfo::adj_rhs, ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::b, arm_compute::test::validation::data_type, MatMulKernelInfo::export_rhs_to_cl_image, and ClMatMulNativeKernel::validate().