Compute Library
 21.08
store.h
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2018-2020 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #ifndef ARM_COMPUTE_WRAPPER_STORE_H
25 #define ARM_COMPUTE_WRAPPER_STORE_H
26 
27 #include <arm_neon.h>
28 
29 namespace arm_compute
30 {
31 namespace wrapper
32 {
33 #define VSTORE_IMPL(stype, vtype, prefix, postfix) \
34  inline void vstore(stype *ptr, vtype val) \
35  { \
36  prefix##_##postfix(ptr, val); \
37  }
38 
39 VSTORE_IMPL(uint8_t, uint8x8_t, vst1, u8)
40 VSTORE_IMPL(uint8_t, uint8x8x2_t, vst2, u8)
41 VSTORE_IMPL(int8_t, int8x8_t, vst1, s8)
42 VSTORE_IMPL(int8_t, int8x8x2_t, vst2, s8)
43 VSTORE_IMPL(uint16_t, uint16x4_t, vst1, u16)
44 VSTORE_IMPL(int16_t, int16x4_t, vst1, s16)
45 VSTORE_IMPL(uint32_t, uint32x2_t, vst1, u32)
46 VSTORE_IMPL(int32_t, int32x2_t, vst1, s32)
47 //VSTORE_IMPL(uint64_t, 1, vst1, u64)
48 //VSTORE_IMPL(int64_t, 1, vst1, s64)
49 VSTORE_IMPL(float, float32x2_t, vst1, f32)
50 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
51 VSTORE_IMPL(float16_t, float16x4_t, vst1, f16)
52 #endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
53 
54 VSTORE_IMPL(uint8_t, uint8x16_t, vst1q, u8)
55 VSTORE_IMPL(int8_t, int8x16_t, vst1q, s8)
56 VSTORE_IMPL(uint16_t, uint16x8_t, vst1q, u16)
57 VSTORE_IMPL(int16_t, int16x8_t, vst1q, s16)
58 VSTORE_IMPL(uint32_t, uint32x4_t, vst1q, u32)
59 VSTORE_IMPL(int32_t, int32x4_t, vst1q, s32)
60 //VSTORE_IMPL(uint64_t, 2, vst1q, u64)
61 //VSTORE_IMPL(int64_t, 2, vst1q, s64)
62 VSTORE_IMPL(float, float32x4_t, vst1q, f32)
63 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
64 VSTORE_IMPL(float16_t, float16x8_t, vst1q, f16)
65 #endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
66 
67 #undef VSTORE_IMPL
68 } // namespace wrapper
69 } // namespace arm_compute
70 #endif /* ARM_COMPUTE_WRAPPER_STORE_H */
#define VSTORE_IMPL(stype, vtype, prefix, postfix)
Definition: store.h:33
Copyright (c) 2017-2021 Arm Limited.