arm_fp16.h 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579
  1. /* ARM FP16 scalar 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 _AARCH64_FP16_H_
  21. #define _AARCH64_FP16_H_
  22. #include <stdint.h>
  23. #pragma GCC push_options
  24. #pragma GCC target ("arch=armv8.2-a+fp16")
  25. typedef __fp16 float16_t;
  26. /* ARMv8.2-A FP16 one operand scalar intrinsics. */
  27. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  28. vabsh_f16 (float16_t __a)
  29. {
  30. return __builtin_aarch64_abshf (__a);
  31. }
  32. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  33. vceqzh_f16 (float16_t __a)
  34. {
  35. return __builtin_aarch64_cmeqhf_uss (__a, 0.0f);
  36. }
  37. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  38. vcgezh_f16 (float16_t __a)
  39. {
  40. return __builtin_aarch64_cmgehf_uss (__a, 0.0f);
  41. }
  42. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  43. vcgtzh_f16 (float16_t __a)
  44. {
  45. return __builtin_aarch64_cmgthf_uss (__a, 0.0f);
  46. }
  47. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  48. vclezh_f16 (float16_t __a)
  49. {
  50. return __builtin_aarch64_cmlehf_uss (__a, 0.0f);
  51. }
  52. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  53. vcltzh_f16 (float16_t __a)
  54. {
  55. return __builtin_aarch64_cmlthf_uss (__a, 0.0f);
  56. }
  57. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  58. vcvth_f16_s16 (int16_t __a)
  59. {
  60. return __builtin_aarch64_floathihf (__a);
  61. }
  62. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  63. vcvth_f16_s32 (int32_t __a)
  64. {
  65. return __builtin_aarch64_floatsihf (__a);
  66. }
  67. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  68. vcvth_f16_s64 (int64_t __a)
  69. {
  70. return __builtin_aarch64_floatdihf (__a);
  71. }
  72. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  73. vcvth_f16_u16 (uint16_t __a)
  74. {
  75. return __builtin_aarch64_floatunshihf_us (__a);
  76. }
  77. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  78. vcvth_f16_u32 (uint32_t __a)
  79. {
  80. return __builtin_aarch64_floatunssihf_us (__a);
  81. }
  82. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  83. vcvth_f16_u64 (uint64_t __a)
  84. {
  85. return __builtin_aarch64_floatunsdihf_us (__a);
  86. }
  87. __extension__ static __inline int16_t __attribute__ ((__always_inline__))
  88. vcvth_s16_f16 (float16_t __a)
  89. {
  90. return __builtin_aarch64_fix_trunchfhi (__a);
  91. }
  92. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  93. vcvth_s32_f16 (float16_t __a)
  94. {
  95. return __builtin_aarch64_fix_trunchfsi (__a);
  96. }
  97. __extension__ static __inline int64_t __attribute__ ((__always_inline__))
  98. vcvth_s64_f16 (float16_t __a)
  99. {
  100. return __builtin_aarch64_fix_trunchfdi (__a);
  101. }
  102. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  103. vcvth_u16_f16 (float16_t __a)
  104. {
  105. return __builtin_aarch64_fixuns_trunchfhi_us (__a);
  106. }
  107. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  108. vcvth_u32_f16 (float16_t __a)
  109. {
  110. return __builtin_aarch64_fixuns_trunchfsi_us (__a);
  111. }
  112. __extension__ static __inline uint64_t __attribute__ ((__always_inline__))
  113. vcvth_u64_f16 (float16_t __a)
  114. {
  115. return __builtin_aarch64_fixuns_trunchfdi_us (__a);
  116. }
  117. __extension__ static __inline int16_t __attribute__ ((__always_inline__))
  118. vcvtah_s16_f16 (float16_t __a)
  119. {
  120. return __builtin_aarch64_lroundhfhi (__a);
  121. }
  122. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  123. vcvtah_s32_f16 (float16_t __a)
  124. {
  125. return __builtin_aarch64_lroundhfsi (__a);
  126. }
  127. __extension__ static __inline int64_t __attribute__ ((__always_inline__))
  128. vcvtah_s64_f16 (float16_t __a)
  129. {
  130. return __builtin_aarch64_lroundhfdi (__a);
  131. }
  132. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  133. vcvtah_u16_f16 (float16_t __a)
  134. {
  135. return __builtin_aarch64_lrounduhfhi_us (__a);
  136. }
  137. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  138. vcvtah_u32_f16 (float16_t __a)
  139. {
  140. return __builtin_aarch64_lrounduhfsi_us (__a);
  141. }
  142. __extension__ static __inline uint64_t __attribute__ ((__always_inline__))
  143. vcvtah_u64_f16 (float16_t __a)
  144. {
  145. return __builtin_aarch64_lrounduhfdi_us (__a);
  146. }
  147. __extension__ static __inline int16_t __attribute__ ((__always_inline__))
  148. vcvtmh_s16_f16 (float16_t __a)
  149. {
  150. return __builtin_aarch64_lfloorhfhi (__a);
  151. }
  152. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  153. vcvtmh_s32_f16 (float16_t __a)
  154. {
  155. return __builtin_aarch64_lfloorhfsi (__a);
  156. }
  157. __extension__ static __inline int64_t __attribute__ ((__always_inline__))
  158. vcvtmh_s64_f16 (float16_t __a)
  159. {
  160. return __builtin_aarch64_lfloorhfdi (__a);
  161. }
  162. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  163. vcvtmh_u16_f16 (float16_t __a)
  164. {
  165. return __builtin_aarch64_lflooruhfhi_us (__a);
  166. }
  167. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  168. vcvtmh_u32_f16 (float16_t __a)
  169. {
  170. return __builtin_aarch64_lflooruhfsi_us (__a);
  171. }
  172. __extension__ static __inline uint64_t __attribute__ ((__always_inline__))
  173. vcvtmh_u64_f16 (float16_t __a)
  174. {
  175. return __builtin_aarch64_lflooruhfdi_us (__a);
  176. }
  177. __extension__ static __inline int16_t __attribute__ ((__always_inline__))
  178. vcvtnh_s16_f16 (float16_t __a)
  179. {
  180. return __builtin_aarch64_lfrintnhfhi (__a);
  181. }
  182. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  183. vcvtnh_s32_f16 (float16_t __a)
  184. {
  185. return __builtin_aarch64_lfrintnhfsi (__a);
  186. }
  187. __extension__ static __inline int64_t __attribute__ ((__always_inline__))
  188. vcvtnh_s64_f16 (float16_t __a)
  189. {
  190. return __builtin_aarch64_lfrintnhfdi (__a);
  191. }
  192. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  193. vcvtnh_u16_f16 (float16_t __a)
  194. {
  195. return __builtin_aarch64_lfrintnuhfhi_us (__a);
  196. }
  197. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  198. vcvtnh_u32_f16 (float16_t __a)
  199. {
  200. return __builtin_aarch64_lfrintnuhfsi_us (__a);
  201. }
  202. __extension__ static __inline uint64_t __attribute__ ((__always_inline__))
  203. vcvtnh_u64_f16 (float16_t __a)
  204. {
  205. return __builtin_aarch64_lfrintnuhfdi_us (__a);
  206. }
  207. __extension__ static __inline int16_t __attribute__ ((__always_inline__))
  208. vcvtph_s16_f16 (float16_t __a)
  209. {
  210. return __builtin_aarch64_lceilhfhi (__a);
  211. }
  212. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  213. vcvtph_s32_f16 (float16_t __a)
  214. {
  215. return __builtin_aarch64_lceilhfsi (__a);
  216. }
  217. __extension__ static __inline int64_t __attribute__ ((__always_inline__))
  218. vcvtph_s64_f16 (float16_t __a)
  219. {
  220. return __builtin_aarch64_lceilhfdi (__a);
  221. }
  222. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  223. vcvtph_u16_f16 (float16_t __a)
  224. {
  225. return __builtin_aarch64_lceiluhfhi_us (__a);
  226. }
  227. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  228. vcvtph_u32_f16 (float16_t __a)
  229. {
  230. return __builtin_aarch64_lceiluhfsi_us (__a);
  231. }
  232. __extension__ static __inline uint64_t __attribute__ ((__always_inline__))
  233. vcvtph_u64_f16 (float16_t __a)
  234. {
  235. return __builtin_aarch64_lceiluhfdi_us (__a);
  236. }
  237. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  238. vnegh_f16 (float16_t __a)
  239. {
  240. return __builtin_aarch64_neghf (__a);
  241. }
  242. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  243. vrecpeh_f16 (float16_t __a)
  244. {
  245. return __builtin_aarch64_frecpehf (__a);
  246. }
  247. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  248. vrecpxh_f16 (float16_t __a)
  249. {
  250. return __builtin_aarch64_frecpxhf (__a);
  251. }
  252. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  253. vrndh_f16 (float16_t __a)
  254. {
  255. return __builtin_aarch64_btrunchf (__a);
  256. }
  257. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  258. vrndah_f16 (float16_t __a)
  259. {
  260. return __builtin_aarch64_roundhf (__a);
  261. }
  262. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  263. vrndih_f16 (float16_t __a)
  264. {
  265. return __builtin_aarch64_nearbyinthf (__a);
  266. }
  267. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  268. vrndmh_f16 (float16_t __a)
  269. {
  270. return __builtin_aarch64_floorhf (__a);
  271. }
  272. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  273. vrndnh_f16 (float16_t __a)
  274. {
  275. return __builtin_aarch64_frintnhf (__a);
  276. }
  277. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  278. vrndph_f16 (float16_t __a)
  279. {
  280. return __builtin_aarch64_ceilhf (__a);
  281. }
  282. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  283. vrndxh_f16 (float16_t __a)
  284. {
  285. return __builtin_aarch64_rinthf (__a);
  286. }
  287. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  288. vrsqrteh_f16 (float16_t __a)
  289. {
  290. return __builtin_aarch64_rsqrtehf (__a);
  291. }
  292. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  293. vsqrth_f16 (float16_t __a)
  294. {
  295. return __builtin_aarch64_sqrthf (__a);
  296. }
  297. /* ARMv8.2-A FP16 two operands scalar intrinsics. */
  298. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  299. vaddh_f16 (float16_t __a, float16_t __b)
  300. {
  301. return __a + __b;
  302. }
  303. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  304. vabdh_f16 (float16_t __a, float16_t __b)
  305. {
  306. return __builtin_aarch64_fabdhf (__a, __b);
  307. }
  308. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  309. vcageh_f16 (float16_t __a, float16_t __b)
  310. {
  311. return __builtin_aarch64_facgehf_uss (__a, __b);
  312. }
  313. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  314. vcagth_f16 (float16_t __a, float16_t __b)
  315. {
  316. return __builtin_aarch64_facgthf_uss (__a, __b);
  317. }
  318. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  319. vcaleh_f16 (float16_t __a, float16_t __b)
  320. {
  321. return __builtin_aarch64_faclehf_uss (__a, __b);
  322. }
  323. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  324. vcalth_f16 (float16_t __a, float16_t __b)
  325. {
  326. return __builtin_aarch64_faclthf_uss (__a, __b);
  327. }
  328. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  329. vceqh_f16 (float16_t __a, float16_t __b)
  330. {
  331. return __builtin_aarch64_cmeqhf_uss (__a, __b);
  332. }
  333. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  334. vcgeh_f16 (float16_t __a, float16_t __b)
  335. {
  336. return __builtin_aarch64_cmgehf_uss (__a, __b);
  337. }
  338. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  339. vcgth_f16 (float16_t __a, float16_t __b)
  340. {
  341. return __builtin_aarch64_cmgthf_uss (__a, __b);
  342. }
  343. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  344. vcleh_f16 (float16_t __a, float16_t __b)
  345. {
  346. return __builtin_aarch64_cmlehf_uss (__a, __b);
  347. }
  348. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  349. vclth_f16 (float16_t __a, float16_t __b)
  350. {
  351. return __builtin_aarch64_cmlthf_uss (__a, __b);
  352. }
  353. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  354. vcvth_n_f16_s16 (int16_t __a, const int __b)
  355. {
  356. return __builtin_aarch64_scvtfhi (__a, __b);
  357. }
  358. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  359. vcvth_n_f16_s32 (int32_t __a, const int __b)
  360. {
  361. return __builtin_aarch64_scvtfsihf (__a, __b);
  362. }
  363. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  364. vcvth_n_f16_s64 (int64_t __a, const int __b)
  365. {
  366. return __builtin_aarch64_scvtfdihf (__a, __b);
  367. }
  368. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  369. vcvth_n_f16_u16 (uint16_t __a, const int __b)
  370. {
  371. return __builtin_aarch64_ucvtfhi_sus (__a, __b);
  372. }
  373. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  374. vcvth_n_f16_u32 (uint32_t __a, const int __b)
  375. {
  376. return __builtin_aarch64_ucvtfsihf_sus (__a, __b);
  377. }
  378. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  379. vcvth_n_f16_u64 (uint64_t __a, const int __b)
  380. {
  381. return __builtin_aarch64_ucvtfdihf_sus (__a, __b);
  382. }
  383. __extension__ static __inline int16_t __attribute__ ((__always_inline__))
  384. vcvth_n_s16_f16 (float16_t __a, const int __b)
  385. {
  386. return __builtin_aarch64_fcvtzshf (__a, __b);
  387. }
  388. __extension__ static __inline int32_t __attribute__ ((__always_inline__))
  389. vcvth_n_s32_f16 (float16_t __a, const int __b)
  390. {
  391. return __builtin_aarch64_fcvtzshfsi (__a, __b);
  392. }
  393. __extension__ static __inline int64_t __attribute__ ((__always_inline__))
  394. vcvth_n_s64_f16 (float16_t __a, const int __b)
  395. {
  396. return __builtin_aarch64_fcvtzshfdi (__a, __b);
  397. }
  398. __extension__ static __inline uint16_t __attribute__ ((__always_inline__))
  399. vcvth_n_u16_f16 (float16_t __a, const int __b)
  400. {
  401. return __builtin_aarch64_fcvtzuhf_uss (__a, __b);
  402. }
  403. __extension__ static __inline uint32_t __attribute__ ((__always_inline__))
  404. vcvth_n_u32_f16 (float16_t __a, const int __b)
  405. {
  406. return __builtin_aarch64_fcvtzuhfsi_uss (__a, __b);
  407. }
  408. __extension__ static __inline uint64_t __attribute__ ((__always_inline__))
  409. vcvth_n_u64_f16 (float16_t __a, const int __b)
  410. {
  411. return __builtin_aarch64_fcvtzuhfdi_uss (__a, __b);
  412. }
  413. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  414. vdivh_f16 (float16_t __a, float16_t __b)
  415. {
  416. return __a / __b;
  417. }
  418. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  419. vmaxh_f16 (float16_t __a, float16_t __b)
  420. {
  421. return __builtin_aarch64_fmaxhf (__a, __b);
  422. }
  423. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  424. vmaxnmh_f16 (float16_t __a, float16_t __b)
  425. {
  426. return __builtin_aarch64_fmaxhf (__a, __b);
  427. }
  428. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  429. vminh_f16 (float16_t __a, float16_t __b)
  430. {
  431. return __builtin_aarch64_fminhf (__a, __b);
  432. }
  433. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  434. vminnmh_f16 (float16_t __a, float16_t __b)
  435. {
  436. return __builtin_aarch64_fminhf (__a, __b);
  437. }
  438. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  439. vmulh_f16 (float16_t __a, float16_t __b)
  440. {
  441. return __a * __b;
  442. }
  443. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  444. vmulxh_f16 (float16_t __a, float16_t __b)
  445. {
  446. return __builtin_aarch64_fmulxhf (__a, __b);
  447. }
  448. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  449. vrecpsh_f16 (float16_t __a, float16_t __b)
  450. {
  451. return __builtin_aarch64_frecpshf (__a, __b);
  452. }
  453. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  454. vrsqrtsh_f16 (float16_t __a, float16_t __b)
  455. {
  456. return __builtin_aarch64_rsqrtshf (__a, __b);
  457. }
  458. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  459. vsubh_f16 (float16_t __a, float16_t __b)
  460. {
  461. return __a - __b;
  462. }
  463. /* ARMv8.2-A FP16 three operands scalar intrinsics. */
  464. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  465. vfmah_f16 (float16_t __a, float16_t __b, float16_t __c)
  466. {
  467. return __builtin_aarch64_fmahf (__b, __c, __a);
  468. }
  469. __extension__ static __inline float16_t __attribute__ ((__always_inline__))
  470. vfmsh_f16 (float16_t __a, float16_t __b, float16_t __c)
  471. {
  472. return __builtin_aarch64_fnmahf (__b, __c, __a);
  473. }
  474. #pragma GCC pop_options
  475. #endif