You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

256 lines
6.6KB

  1. /* ARM FP16 intrinsics include file.
  2. Copyright (C) 2016-2020 Free Software Foundation, Inc.
  3. Contributed by ARM Ltd.
  4. This file is part of GCC.
  5. GCC is free software; you can redistribute it and/or modify it
  6. under the terms of the GNU General Public License as published
  7. by the Free Software Foundation; either version 3, or (at your
  8. option) any later version.
  9. GCC is distributed in the hope that it will be useful, but WITHOUT
  10. ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11. or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
  12. License for more details.
  13. Under Section 7 of GPL version 3, you are granted additional
  14. permissions described in the GCC Runtime Library Exception, version
  15. 3.1, as published by the Free Software Foundation.
  16. You should have received a copy of the GNU General Public License and
  17. a copy of the GCC Runtime Library Exception along with this program;
  18. see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
  19. <http://www.gnu.org/licenses/>. */
  20. #ifndef _GCC_ARM_FP16_H
  21. #define _GCC_ARM_FP16_H 1
  22. #ifdef __cplusplus
  23. extern "C" {
  24. #endif
  25. #include <stdint.h>
  26. /* Intrinsics for FP16 instructions. */
  27. #pragma GCC push_options
  28. #pragma GCC target ("fpu=fp-armv8")
  29. #if defined (__ARM_FEATURE_FP16_SCALAR_ARITHMETIC)
  30. typedef __fp16 float16_t;
  31. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  32. vabsh_f16 (float16_t __a)
  33. {
  34. return __builtin_neon_vabshf (__a);
  35. }
  36. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  37. vaddh_f16 (float16_t __a, float16_t __b)
  38. {
  39. return __a + __b;
  40. }
  41. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  42. vcvtah_s32_f16 (float16_t __a)
  43. {
  44. return __builtin_neon_vcvtahssi (__a);
  45. }
  46. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  47. vcvtah_u32_f16 (float16_t __a)
  48. {
  49. return __builtin_neon_vcvtahusi (__a);
  50. }
  51. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  52. vcvth_f16_s32 (int32_t __a)
  53. {
  54. return __builtin_neon_vcvthshf (__a);
  55. }
  56. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  57. vcvth_f16_u32 (uint32_t __a)
  58. {
  59. return __builtin_neon_vcvthuhf (__a);
  60. }
  61. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  62. vcvth_n_f16_s32 (int32_t __a, const int __b)
  63. {
  64. return __builtin_neon_vcvths_nhf (__a, __b);
  65. }
  66. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  67. vcvth_n_f16_u32 (uint32_t __a, const int __b)
  68. {
  69. return __builtin_neon_vcvthu_nhf ((int32_t)__a, __b);
  70. }
  71. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  72. vcvth_n_s32_f16 (float16_t __a, const int __b)
  73. {
  74. return __builtin_neon_vcvths_nsi (__a, __b);
  75. }
  76. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  77. vcvth_n_u32_f16 (float16_t __a, const int __b)
  78. {
  79. return (uint32_t)__builtin_neon_vcvthu_nsi (__a, __b);
  80. }
  81. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  82. vcvth_s32_f16 (float16_t __a)
  83. {
  84. return __builtin_neon_vcvthssi (__a);
  85. }
  86. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  87. vcvth_u32_f16 (float16_t __a)
  88. {
  89. return __builtin_neon_vcvthusi (__a);
  90. }
  91. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  92. vcvtmh_s32_f16 (float16_t __a)
  93. {
  94. return __builtin_neon_vcvtmhssi (__a);
  95. }
  96. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  97. vcvtmh_u32_f16 (float16_t __a)
  98. {
  99. return __builtin_neon_vcvtmhusi (__a);
  100. }
  101. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  102. vcvtnh_s32_f16 (float16_t __a)
  103. {
  104. return __builtin_neon_vcvtnhssi (__a);
  105. }
  106. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  107. vcvtnh_u32_f16 (float16_t __a)
  108. {
  109. return __builtin_neon_vcvtnhusi (__a);
  110. }
  111. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  112. vcvtph_s32_f16 (float16_t __a)
  113. {
  114. return __builtin_neon_vcvtphssi (__a);
  115. }
  116. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  117. vcvtph_u32_f16 (float16_t __a)
  118. {
  119. return __builtin_neon_vcvtphusi (__a);
  120. }
  121. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  122. vdivh_f16 (float16_t __a, float16_t __b)
  123. {
  124. return __a / __b;
  125. }
  126. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  127. vfmah_f16 (float16_t __a, float16_t __b, float16_t __c)
  128. {
  129. return __builtin_neon_vfmahf (__a, __b, __c);
  130. }
  131. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  132. vfmsh_f16 (float16_t __a, float16_t __b, float16_t __c)
  133. {
  134. return __builtin_neon_vfmshf (__a, __b, __c);
  135. }
  136. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  137. vmaxnmh_f16 (float16_t __a, float16_t __b)
  138. {
  139. return __builtin_neon_vmaxnmhf (__a, __b);
  140. }
  141. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  142. vminnmh_f16 (float16_t __a, float16_t __b)
  143. {
  144. return __builtin_neon_vminnmhf (__a, __b);
  145. }
  146. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  147. vmulh_f16 (float16_t __a, float16_t __b)
  148. {
  149. return __a * __b;
  150. }
  151. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  152. vnegh_f16 (float16_t __a)
  153. {
  154. return - __a;
  155. }
  156. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  157. vrndah_f16 (float16_t __a)
  158. {
  159. return __builtin_neon_vrndahf (__a);
  160. }
  161. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  162. vrndh_f16 (float16_t __a)
  163. {
  164. return __builtin_neon_vrndhf (__a);
  165. }
  166. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  167. vrndih_f16 (float16_t __a)
  168. {
  169. return __builtin_neon_vrndihf (__a);
  170. }
  171. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  172. vrndmh_f16 (float16_t __a)
  173. {
  174. return __builtin_neon_vrndmhf (__a);
  175. }
  176. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  177. vrndnh_f16 (float16_t __a)
  178. {
  179. return __builtin_neon_vrndnhf (__a);
  180. }
  181. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  182. vrndph_f16 (float16_t __a)
  183. {
  184. return __builtin_neon_vrndphf (__a);
  185. }
  186. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  187. vrndxh_f16 (float16_t __a)
  188. {
  189. return __builtin_neon_vrndxhf (__a);
  190. }
  191. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  192. vsqrth_f16 (float16_t __a)
  193. {
  194. return __builtin_neon_vsqrthf (__a);
  195. }
  196. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  197. vsubh_f16 (float16_t __a, float16_t __b)
  198. {
  199. return __a - __b;
  200. }
  201. #endif /* __ARM_FEATURE_FP16_SCALAR_ARITHMETIC */
  202. #pragma GCC pop_options
  203. #ifdef __cplusplus
  204. }
  205. #endif
  206. #endif