diff options
Diffstat (limited to 'src/cpu/kernels/CpuKernelSelectionTypes.h')
| -rw-r--r-- | src/cpu/kernels/CpuKernelSelectionTypes.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuKernelSelectionTypes.h b/src/cpu/kernels/CpuKernelSelectionTypes.h index 45ebeec394..7c1e4772a6 100644 --- a/src/cpu/kernels/CpuKernelSelectionTypes.h +++ b/src/cpu/kernels/CpuKernelSelectionTypes.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021-2023 Arm Limited. + * Copyright (c) 2021-2024 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -104,6 +104,8 @@ struct SoftmaxKernelDataTypeISASelectorData DataType dt; cpuinfo::CpuIsaInfo isa; bool is_log; + int axis; + unsigned long sme2_vector_length; }; // Selector pointer types |