Line |
Branch |
Exec |
Source |
1 |
|
|
// |
2 |
|
|
// SPDX-FileCopyrightText: Copyright 2024 Arm Limited and/or its affiliates <open-source-office@arm.com> |
3 |
|
|
// |
4 |
|
|
// SPDX-License-Identifier: Apache-2.0 |
5 |
|
|
// |
6 |
|
|
|
7 |
|
|
#pragma once |
8 |
|
|
|
9 |
|
|
#include <cstddef> |
10 |
|
|
#include <cstdint> |
11 |
|
|
|
12 |
|
|
namespace kai::test { |
13 |
|
|
|
14 |
|
|
/// Gets the SME vector length. |
15 |
|
|
template <size_t esize> |
16 |
|
|
uint64_t get_sme_vector_length(); |
17 |
|
|
|
18 |
|
|
/// Gets the SME vector length. |
19 |
|
|
template <typename T> |
20 |
|
172 |
uint64_t get_sme_vector_length() { |
21 |
|
172 |
return get_sme_vector_length<sizeof(T)>(); |
22 |
|
|
} |
23 |
|
|
|
24 |
|
|
} // namespace kai::test |
25 |
|
|
|