24.02.1
|
Go to the documentation of this file.
49 unsigned int m,
unsigned int n,
unsigned int k,
unsigned int b,
bool is_rhs_constant);
52 static std::map<DataType, FunctionExecutorPtr> gemm_configs = {
62 if (gemm_configs.find(
data_type) != gemm_configs.end())
71 unsigned int m,
unsigned int n,
unsigned int k,
unsigned int b,
bool is_rhs_constant)
80 unsigned int m,
unsigned int n,
unsigned int k,
unsigned int b,
bool is_rhs_constant)
89 unsigned int m,
unsigned int n,
unsigned int k,
unsigned int b,
bool is_rhs_constant)
@ QSYMM8_PER_CHANNEL
quantized, symmetric per channel fixed-point 8-bit number
OpenCL GEMM kernel selection parameters.
DataType data_type
Data type.
@ QASYMM8
quantized, asymmetric fixed-point 8-bit number unsigned
@ QSYMM8
quantized, symmetric fixed-point 8-bit number
#define ARM_COMPUTE_ERROR(msg)
Print the given message then throw an std::runtime_error.
unsigned int n
Number of columns for the rhs matrix.
Midgard based OpenCL GEMMKernel selection.
@ NATIVE
Native GEMM kernel with configurable block size.
CLGEMMDefaultTypeMidgard(GPUTarget gpu)
Constructor.
unsigned int m
Number of rows for the lhs matrix.
Manages all the OpenCL kernels compilation and caching, provides accessors for the OpenCL Context.
bool is_rhs_constant
True if the content of the rhs matrix is constant.
CLGEMMKernelType
OpenCL GEMM kernel types.
@ QASYMM8_SIGNED
quantized, asymmetric fixed-point 8-bit number signed
unsigned int b
Batch size.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Basic interface for the GEMM kernel selection.
GPUTarget
Available GPU Targets.
unsigned int k
Number of rows for the rhs matrix.
Copyright (c) 2017-2024 Arm Limited.
@ F16
16-bit floating-point number
@ RESHAPED
Reshaped GEMM kernel where both lhs and rhs matrices are reshaped.
CLGEMMKernelType select_kernel(const CLGEMMKernelSelectionParams ¶ms) override
Given the input parameters passed through CLGEMMKernelSelectionParams, this method returns the CLGEMM...
@ F32
32-bit floating-point number
DataType
Available data types.