Compute Library
 21.05
CLGEMMDefaultConfigReshapedBifrost Class Referencefinal

Bifrost based OpenCL GEMMReshaped configuration. More...

#include <CLGEMMDefaultConfigReshapedBifrost.h>

Collaboration diagram for CLGEMMDefaultConfigReshapedBifrost:
[legend]

Public Member Functions

 CLGEMMDefaultConfigReshapedBifrost (GPUTarget gpu)
 Constructor. More...
 
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfoconfigure (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 ICLGEMMKernelConfiguration
 ICLGEMMKernelConfiguration (GPUTarget arch)
 Constructor. More...
 
 ICLGEMMKernelConfiguration (const ICLGEMMKernelConfiguration &)=delete
 Prevent instances of this class from being copied (As this class contains pointers) More...
 
ICLGEMMKernelConfigurationoperator= (const ICLGEMMKernelConfiguration &)=delete
 Prevent instances of this class from being copied (As this class contains pointers) More...
 
 ICLGEMMKernelConfiguration (ICLGEMMKernelConfiguration &&)=default
 Default Move Constructor. More...
 
ICLGEMMKernelConfigurationoperator= (ICLGEMMKernelConfiguration &&)=default
 Default move assignment operator. More...
 
virtual ~ICLGEMMKernelConfiguration ()=default
 Virtual destructor. More...
 

Detailed Description

Bifrost based OpenCL GEMMReshaped configuration.

Definition at line 34 of file CLGEMMDefaultConfigReshapedBifrost.h.

Constructor & Destructor Documentation

◆ CLGEMMDefaultConfigReshapedBifrost()

Constructor.

Parameters
[in]gpuGPU target

Definition at line 42 of file CLGEMMDefaultConfigReshapedBifrost.cpp.

44 {
45 }
ICLGEMMKernelConfiguration(GPUTarget arch)
Constructor.

Member Function Documentation

◆ configure()

std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > configure ( unsigned int  m,
unsigned int  n,
unsigned int  k,
unsigned int  b,
DataType  data_type 
)
overridevirtual

Given M, N, K and B, this method returns the GEMMLHSMatrixInfo and GEMMRHSMatrixInfo to be used.

Parameters
[in]mNumber of rows LHS matrix
[in]nNumber of columns RHS matrix
[in]kNumber of columns LHS matrix or number of rows RHS matrix
[in]bBatch size
[in]data_typeData type

Implements ICLGEMMKernelConfiguration.

Definition at line 47 of file CLGEMMDefaultConfigReshapedBifrost.cpp.

48 {
49  using ConfigurationFunctionExecutorPtr = std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> (CLGEMMDefaultConfigReshapedBifrost::*)(unsigned int m, unsigned int n, unsigned int k, unsigned int b);
50 
51  CLGEMMConfigArray<ConfigurationFunctionExecutorPtr> configs_G7x(&CLGEMMDefaultConfigReshapedBifrost::configure_G7x_f32,
52  &CLGEMMDefaultConfigReshapedBifrost::configure_G7x_f16,
53  &CLGEMMDefaultConfigReshapedBifrost::configure_G7x_u8);
54 
55  CLGEMMConfigArray<ConfigurationFunctionExecutorPtr> configs_G52(&CLGEMMDefaultConfigReshapedBifrost::configure_G52_f32,
56  &CLGEMMDefaultConfigReshapedBifrost::configure_G52_f16,
57  &CLGEMMDefaultConfigReshapedBifrost::configure_G7x_u8);
58 
59  CLGEMMConfigArray<ConfigurationFunctionExecutorPtr> configs_G76(&CLGEMMDefaultConfigReshapedBifrost::configure_G76_f32,
60  &CLGEMMDefaultConfigReshapedBifrost::configure_G76_f16,
61  &CLGEMMDefaultConfigReshapedBifrost::configure_G76_u8);
62 
63  ConfigurationFunctionExecutorPtr func = nullptr;
64 
65  switch(_target)
66  {
67  case GPUTarget::G76:
68  func = configs_G76.get_function(data_type);
69  break;
70  case GPUTarget::G52:
71  func = configs_G52.get_function(data_type);
72  break;
73  default:
74  func = configs_G7x.get_function(data_type);
75  break;
76  }
77 
78  ARM_COMPUTE_ERROR_ON_MSG(func == nullptr, "Data type not support for GEMM");
79  return (this->*func)(m, n, k, b);
80 }
SimpleTensor< float > b
Definition: DFT.cpp:157
const DataType data_type
Definition: Im2Col.cpp:150
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
Definition: Error.h:456
FloorUKernelPtr func

References ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::b, arm_compute::test::validation::data_type, func, arm_compute::G52, arm_compute::G76, and CLGEMMConfigArray< T >::get_function().


The documentation for this class was generated from the following files: