diff options
| author | Omar Al Khatib <omar.alkhatib@arm.com> | 2024-04-23 16:26:56 +0100 |
|---|---|---|
| committer | Omar Al Khatib <omar.alkhatib@arm.com> | 2024-05-08 12:16:10 +0000 |
| commit | c1575b2c12a4cee3a60c711fe6521025a814b159 (patch) | |
| tree | dd88487711f631a83989dec5c7e058d29cc076ec /arm_compute/core/CPP/CPPTypes.h | |
| parent | 2fea13593a4753316ae488edf489cb4b00150153 (diff) | |
| download | ComputeLibrary-c1575b2c12a4cee3a60c711fe6521025a814b159.tar.gz | |
Add SME2 implementation of Softmax for QASYMM8 and QASYMM8_SIGNED.
Resolves: [COMPMID-6917]
Change-Id: Id8b96efd29f6c61dd43a371341c6e1fe087953e9
Signed-off-by: Omar Al Khatib <omar.alkhatib@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11509
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/CPP/CPPTypes.h')
| -rw-r--r-- | arm_compute/core/CPP/CPPTypes.h | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/arm_compute/core/CPP/CPPTypes.h b/arm_compute/core/CPP/CPPTypes.h index b080a86938..139d630fd7 100644 --- a/arm_compute/core/CPP/CPPTypes.h +++ b/arm_compute/core/CPP/CPPTypes.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2022 Arm Limited. + * Copyright (c) 2017-2022, 2024 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -21,8 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#ifndef ARM_COMPUTE_CPP_TYPES_H -#define ARM_COMPUTE_CPP_TYPES_H +#ifndef ACL_ARM_COMPUTE_CORE_CPP_CPPTYPES_H +#define ACL_ARM_COMPUTE_CORE_CPP_CPPTYPES_H #include "arm_compute/core/Error.h" @@ -171,6 +171,12 @@ public: */ unsigned int get_cpu_num() const; + /** Return the vector length in bytes for sme2 + * + * @return Vector length if sme2 is enabled, otherwise returns 0. + */ + unsigned long get_sme2_vector_length() const; + private: struct Impl; std::unique_ptr<Impl> _impl; @@ -184,4 +190,4 @@ struct ThreadInfo const CPUInfo *cpu_info{nullptr}; }; } // namespace arm_compute -#endif /* ARM_COMPUTE_CPP_TYPES_H */ +#endif // ACL_ARM_COMPUTE_CORE_CPP_CPPTYPES_H |