aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/CpuKernelSelectionTypes.h
diff options
context:
space:
mode:
authoralerah01 <alex.rahlis@arm.com>2022-01-31 19:04:10 +0200
committerAlex Rahlis <alex.rahlis@arm.com>2022-02-22 09:26:32 +0000
commitc9e519d2ea4780297d71e68cccc5de9c7bb7c0b4 (patch)
tree933427df414593e4a40d4c269c8065858a635d65 /src/cpu/kernels/CpuKernelSelectionTypes.h
parent0597b7b55ba0c1aa2cc58e80f95dc918b92afe68 (diff)
downloadComputeLibrary-c9e519d2ea4780297d71e68cccc5de9c7bb7c0b4.tar.gz
Decouple CpuDirectConv2dKernel
Resolves COMPMID-4626 Exclude SVE & SVE2 paths from android.bp NDK version does not support these extensions. Change-Id: I49b147d2a84819975d3225f2920106fa1a0d742f Signed-off-by: alerah01 <alex.rahlis@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7136 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.h8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/cpu/kernels/CpuKernelSelectionTypes.h b/src/cpu/kernels/CpuKernelSelectionTypes.h
index 4a0ebd6e3f..8c5a39ad49 100644
--- a/src/cpu/kernels/CpuKernelSelectionTypes.h
+++ b/src/cpu/kernels/CpuKernelSelectionTypes.h
@@ -40,6 +40,13 @@ struct DataTypeISASelectorData
cpuinfo::CpuIsaInfo isa;
};
+struct DataTypeDataLayoutISASelectorData
+{
+ DataType dt;
+ DataLayout dl;
+ const cpuinfo::CpuIsaInfo &isa;
+};
+
struct PoolDataTypeISASelectorData
{
DataType dt;
@@ -63,6 +70,7 @@ struct DepthwiseConv2dNativeDataTypeISASelectorData
};
// Selector pointer types
using DataTypeISASelectorPtr = std::add_pointer<bool(const DataTypeISASelectorData &data)>::type;
+using DataTypeDataLayoutSelectorPtr = std::add_pointer<bool(const DataTypeDataLayoutISASelectorData &data)>::type;
using PoolDataTypeISASelectorPtr = std::add_pointer<bool(const PoolDataTypeISASelectorData &data)>::type;
using ElementwiseDataTypeISASelectorPtr = std::add_pointer<bool(const ElementwiseDataTypeISASelectorData &data)>::type;
using DepthwiseConv2dNativeDataTypeISASelectorPtr = std::add_pointer<bool(const DepthwiseConv2dNativeDataTypeISASelectorData &data)>::type;