24.02.1
|
Basic interface for the GEMM kernel configuration. More...
#include <IClGemmKernelConfig.h>
Public Member Functions | |
IClGemmKernelConfig (GPUTarget arch) | |
Constructor. More... | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (IClGemmKernelConfig) | |
virtual | ~IClGemmKernelConfig ()=default |
Virtual destructor. More... | |
virtual std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > | configure (unsigned int m, unsigned int n, unsigned int k, unsigned int b, DataType data_type)=0 |
Given M, N, K and B, this method returns the GEMMLHSMatrixInfo and GEMMRHSMatrixInfo to be used. More... | |
Basic interface for the GEMM kernel configuration.
Definition at line 92 of file IClGemmKernelConfig.h.
|
inline |
|
virtualdefault |
Virtual destructor.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE | ( | IClGemmKernelConfig | ) |
|
pure virtual |
Given M, N, K and B, this method returns the GEMMLHSMatrixInfo and GEMMRHSMatrixInfo to be used.
[in] | m | Number of rows LHS matrix |
[in] | n | Number of columns RHS matrix |
[in] | k | Number of columns LHS matrix or number of rows RHS matrix |
[in] | b | Batch size |
[in] | data_type | Data type |
Implemented in ClGemmDefaultConfigNativeBifrost, ClGemmDefaultConfigNativeMidgard, ClGemmDefaultConfigNativeValhall, ClGemmDefaultConfigReshapedBifrost, ClGemmDefaultConfigReshapedValhall, ClGemmDefaultConfigReshapedRhsOnlyBifrost, and ClGemmDefaultConfigReshapedRhsOnlyValhall.