diff options
author | Pablo Marquez Tello <pablo.tello@arm.com> | 2024-06-10 15:03:36 +0100 |
---|---|---|
committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2024-06-10 16:02:28 +0000 |
commit | 311753a2ba5f86257936f03c03f0597c968a92c4 (patch) | |
tree | 166697c908fb6937750321143ed6c848c26df247 | |
parent | f1723a0f1fb04ab1dc34f3d36b32f73e8817822d (diff) | |
download | ComputeLibrary-311753a2ba5f86257936f03c03f0597c968a92c4.tar.gz |
Fixed illegal instruction in Softmax
* The softmax kernel is using SME2 instructions on non SME2 devices
* Resolves MLCE-1304
Change-Id: I9d7d94443e7c9df4e7c1a05eeef6838f530b357b
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11676
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
-rw-r--r-- | src/core/CPP/CPPTypes.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/core/CPP/CPPTypes.cpp b/src/core/CPP/CPPTypes.cpp index 67fbce490f..ee39210fa5 100644 --- a/src/core/CPP/CPPTypes.cpp +++ b/src/core/CPP/CPPTypes.cpp @@ -140,7 +140,10 @@ unsigned int CPUInfo::get_L2_cache_size() const unsigned long CPUInfo::get_sme2_vector_length() const { #ifdef ARM_COMPUTE_ENABLE_SME2 - return arm_gemm::utils::sme::get_vector_length<int8_t>(); + if (this->has_sme2()) + return arm_gemm::utils::sme::get_vector_length<int8_t>(); + else + return 0; #else // ARM_COMPUTE_ENABLE_SME2 return 0; #endif // ARM_COMPUTE_ENABLE_SME2 |