diff options
Diffstat (limited to 'src/core/CPP/CPPTypes.cpp')
| -rw-r--r-- | src/core/CPP/CPPTypes.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/core/CPP/CPPTypes.cpp b/src/core/CPP/CPPTypes.cpp index 67fbce490f..a6d08e5bad 100644 --- a/src/core/CPP/CPPTypes.cpp +++ b/src/core/CPP/CPPTypes.cpp @@ -137,10 +137,13 @@ unsigned int CPUInfo::get_L2_cache_size() const return _impl->L2_cache_size; } -unsigned long CPUInfo::get_sme2_vector_length() const +uint64_t 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 |