24.02.1
|
Go to the documentation of this file.
24 #ifndef SRC_RUNTIME_CL_GEMM_AUTO_HEURISTICS_CL_GEMM_AUTO_HEURISTICS_H
25 #define SRC_RUNTIME_CL_GEMM_AUTO_HEURISTICS_CL_GEMM_AUTO_HEURISTICS_H
35 namespace auto_heuristics
136 #endif // SRC_RUNTIME_CL_GEMM_AUTO_HEURISTICS_CL_GEMM_AUTO_HEURISTICS_H
Result of querying about GEMM config ( GEMMLHSMatrixInfo and GEMMRHSMatrixInfo)
CLGEMMKernelType gemm_type
If the result is valid.
GEMMTypeResult select_mlgo_gemm_kernel(const CommonQuery &query, bool reshape_b_only_on_first_run)
Select gemm type based on mlgo heuristics.
GEMMConfigResult select_mlgo_gemm_config_reshaped_only_rhs(const CommonQuery &query)
Select gemm config based on mlgo heuristics.
CLGEMMKernelType
OpenCL GEMM kernel types.
Result of querying about GEMM type ( CLGEMMKernelType)
A collection of adaptor functions that enable the auto selection between mlgo-based heuristics and de...
GEMMConfigResult select_default_gemm_config_reshaped_only_rhs(const CommonQuery &query)
Select gemm config based on default heuristics.
GEMMLHSMatrixInfo lhs_info
If the result is valid.
GEMMConfigResult select_default_gemm_config_reshaped(const CommonQuery &query)
Select gemm config based on default heuristics.
GEMMTypeResult(bool valid, CLGEMMKernelType gemm_type)
unsigned int n
Number of columns for the rhs matrix.
unsigned int k
Number of rows for the rhs matrix.
unsigned int m
Number of rows for the lhs matrix.
GPUTarget gpu_target
Which GPUTarget to query about.
GEMMConfigResult select_mlgo_gemm_config_reshaped(const CommonQuery &query)
Select gemm config based on mlgo heuristics.
GPUTarget
Available GPU Targets.
GEMM LHS (Left Hand Side) matrix information.
unsigned int b
Batch size.
Copyright (c) 2017-2024 Arm Limited.
GEMMRHSMatrixInfo rhs_info
GEMMLHSMatrixInfo
GEMMConfigResult select_default_gemm_config_native(const CommonQuery &query)
Select gemm config based on default heuristics.
GEMMTypeResult select_default_gemm_kernel(const CommonQuery &query, bool reshape_b_only_on_first_run)
Select gemm type based on default heuristics.
DataType data_type
Data type.
GEMM RHS (Right Hand Side) matrix information.
DataType
Available data types.
GEMMConfigResult(bool valid, const GEMMLHSMatrixInfo &lhs_info, const GEMMRHSMatrixInfo &rhs_info)
GEMMConfigResult select_mlgo_gemm_config_native(const CommonQuery &query)
Select gemm config based on mlgo heuristics.