24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CL_GEMM_HELPERS_H
25 #define ARM_COMPUTE_CL_GEMM_HELPERS_H
85 std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo>
87 std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> info_buf,
121 const unsigned int n,
122 const unsigned int k,
123 const unsigned int b,
125 unsigned int &best_m0,
126 unsigned int &best_n0);
138 std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo>
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > configure_lhs_rhs_info(unsigned int m, unsigned int n, unsigned int m0, unsigned int n0, unsigned int k0, unsigned int v0, unsigned int h0, bool lhs_interleave, bool rhs_interleave, bool lhs_transpose, bool rhs_transpose, bool export_to_cl_image)
Configure GEMMLHSMatrixInfo and GEMMRHSMatrixInfo.
bool export_to_cl_image(const ITensorInfo *tensor)
std::vector< std::vector< int32_t > > GeMMConfigsMatrix
UniqueGemmCommon< Top, Tret > gemm(const GemmArgs &args, const OutputStage &os)
Status validate_image2d_support_on_rhs(const ITensorInfo &tensor_reshaped_info, const GEMMRHSMatrixInfo &rhs_info)
Utility function to validate the image2d OpenCL object support on the RHS reshaped matrix.
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > find_lhs_rhs_info(const GeMMConfigsMatrix &configs, unsigned int m, unsigned int n, unsigned int k, unsigned int b)
Find the preferred configurations for the LHS and RHS tensor using the GeMMConfigsMatrix provided by ...
CLTensor * tensor
Pointer to the auxiliary tensor.
void update_padding_for_cl_image(ITensorInfo *tensor)
Update padding required to export the OpenCL buffer to OpenCL image2d.
Copyright (c) 2017-2024 Arm Limited.
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > select_lhs_rhs_info(std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > info_img, std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > info_buf, unsigned int n, unsigned int k, unsigned int b, DataType data_type)
Select GEMMLHSMatrixInfo and GEMMRHSMatrixInfo.
Store the tensor's metadata.
GEMM RHS (Right Hand Side) matrix information.
bool is_mmul_kernel_preferred(const unsigned int m, const unsigned int n, const unsigned int k, const unsigned int b, const DataType data_type, unsigned int &best_m0, unsigned int &best_n0)
Determine if the MMUL kernels should be preferred.
DataType
Available data types.