aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/CpuKernelSelectionTypes.h
diff options
context:
space:
mode:
authorDana Zlotnik <dana.zlotnik@arm.com>2022-01-17 09:54:26 +0200
committerDana Zlotnik <dana.zlotnik@arm.com>2022-02-14 12:49:53 +0000
commit6a2df886f32dcf7af4258163b0652f0fab07ecc5 (patch)
tree4ad16670d54d29de96df7cc5b582d52a6012255a /src/cpu/kernels/CpuKernelSelectionTypes.h
parent69854ba71f91f86c2a1c8a2301e91dcd93030561 (diff)
downloadComputeLibrary-6a2df886f32dcf7af4258163b0652f0fab07ecc5.tar.gz
Add kernel selection UT for submitted kernels
* Softmax kernel * Elementwise unary kernel * Elementwise binary ** This change require some refactor in the kernel cpp and h files Resolves COMPMID-5043 Change-Id: I58979b023ec31d759690847b3f85fc4baefbbf98 Signed-off-by: Dana Zlotnik <dana.zlotnik@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7033 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Diffstat (limited to 'src/cpu/kernels/CpuKernelSelectionTypes.h')
-rw-r--r--src/cpu/kernels/CpuKernelSelectionTypes.h12
1 files changed, 10 insertions, 2 deletions
diff --git a/src/cpu/kernels/CpuKernelSelectionTypes.h b/src/cpu/kernels/CpuKernelSelectionTypes.h
index 60dcea4a35..60bbd5933c 100644
--- a/src/cpu/kernels/CpuKernelSelectionTypes.h
+++ b/src/cpu/kernels/CpuKernelSelectionTypes.h
@@ -49,9 +49,17 @@ struct PoolDataTypeISASelectorData
cpuinfo::CpuIsaInfo isa;
};
+struct ElementwiseDataTypeISASelectorData
+{
+ DataType dt;
+ cpuinfo::CpuIsaInfo isa;
+ int op;
+};
+
// Selector pointer types
-using DataTypeISASelectorPtr = std::add_pointer<bool(const DataTypeISASelectorData &data)>::type;
-using PoolDataTypeISASelectorPtr = std::add_pointer<bool(const PoolDataTypeISASelectorData &data)>::type;
+using DataTypeISASelectorPtr = std::add_pointer<bool(const DataTypeISASelectorData &data)>::type;
+using PoolDataTypeISASelectorPtr = std::add_pointer<bool(const PoolDataTypeISASelectorData &data)>::type;
+using ElementwiseDataTypeISASelectorPtr = std::add_pointer<bool(const ElementwiseDataTypeISASelectorData &data)>::type;
} // namespace kernels
} // namespace cpu