24.02.1
|
Bifrost based OpenCL GEMMReshaped configuration. More...
#include <ClGemmDefaultConfigReshapedBifrost.h>
Public Member Functions | |
ClGemmDefaultConfigReshapedBifrost (GPUTarget gpu) | |
Constructor. More... | |
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > | configure (unsigned int m, unsigned int n, unsigned int k, unsigned int b, DataType data_type) override |
Given M, N, K and B, this method returns the GEMMLHSMatrixInfo and GEMMRHSMatrixInfo to be used. More... | |
Public Member Functions inherited from IClGemmKernelConfig | |
IClGemmKernelConfig (GPUTarget arch) | |
Constructor. More... | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (IClGemmKernelConfig) | |
virtual | ~IClGemmKernelConfig ()=default |
Virtual destructor. More... | |
Bifrost based OpenCL GEMMReshaped configuration.
Definition at line 38 of file ClGemmDefaultConfigReshapedBifrost.h.
Constructor.
[in] | gpu | GPU target |
Definition at line 47 of file ClGemmDefaultConfigReshapedBifrost.cpp.
|
overridevirtual |
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 |
Implements IClGemmKernelConfig.
Definition at line 51 of file ClGemmDefaultConfigReshapedBifrost.cpp.
References ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::b, arm_compute::test::validation::data_type, arm_compute::G52, arm_compute::G76, and CLGEMMConfigArray< T >::get_function().