24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_ICL_GEMM_KERNEL_CONFIG_H
25 #define ARM_COMPUTE_ICL_GEMM_KERNEL_CONFIG_H
60 CLGEMMConfigArray(T func_f32, T func_f16, T func_int8) : _configs{func_f32, func_f16, func_int8}
75 return _configs.at(
DT_F32);
77 return _configs.at(
DT_F16);
88 std::array<T, 3> _configs;
113 virtual std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo>
@ QSYMM8_PER_CHANNEL
quantized, symmetric per channel fixed-point 8-bit number
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(IClGemmKernelConfig)
CLGEMMConfigArray(T func_f32, T func_f16, T func_int8)
Constructor.
UniqueGemmCommon< Top, Tret > gemm(const GemmArgs &args, const OutputStage &os)
static constexpr size_t DT_INT8
Alias for Int8 index.
@ QASYMM8
quantized, asymmetric fixed-point 8-bit number unsigned
Basic interface for the GEMM kernel configuration.
static constexpr size_t DT_F32
Alias for F32 index.
Basic container for the OpenCL GEMM configuration functions.
@ QASYMM8_SIGNED
quantized, asymmetric fixed-point 8-bit number signed
GPUTarget
Available GPU Targets.
static constexpr size_t DT_F16
Alias for F16 index.
Copyright (c) 2017-2024 Arm Limited.
virtual ~IClGemmKernelConfig()=default
Virtual destructor.
@ F16
16-bit floating-point number
@ F32
32-bit floating-point number
IClGemmKernelConfig(GPUTarget arch)
Constructor.
DataType
Available data types.
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.
T get_function(DataType data_type)
Method to return the GEMM configuration function based on data type.