Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | /*===------------- amxfp16intrin.h - AMX_FP16 intrinsics -*- C++ -*---------=== |
| 2 | * |
||
| 3 | * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
| 4 | * See https://llvm.org/LICENSE.txt for license information. |
||
| 5 | * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
| 6 | * |
||
| 7 | *===------------------------------------------------------------------------=== |
||
| 8 | */ |
||
| 9 | |||
| 10 | #ifndef __IMMINTRIN_H |
||
| 11 | #error "Never use <amxfp16intrin.h> directly; use <immintrin.h> instead." |
||
| 12 | #endif /* __IMMINTRIN_H */ |
||
| 13 | |||
| 14 | #ifndef __AMX_FP16INTRIN_H |
||
| 15 | #define __AMX_FP16INTRIN_H |
||
| 16 | #ifdef __x86_64__ |
||
| 17 | |||
| 18 | /// Compute dot-product of FP16 (16-bit) floating-point pairs in tiles \a a |
||
| 19 | /// and \a b, accumulating the intermediate single-precision (32-bit) |
||
| 20 | /// floating-point elements with elements in \a dst, and store the 32-bit |
||
| 21 | /// result back to tile \a dst. |
||
| 22 | /// |
||
| 23 | /// \headerfile <immintrin.h> |
||
| 24 | /// |
||
| 25 | /// \code |
||
| 26 | /// void _tile_dpfp16ps (__tile dst, __tile a, __tile b) |
||
| 27 | /// \endcode |
||
| 28 | /// |
||
| 29 | /// \code{.operation} |
||
| 30 | /// FOR m := 0 TO dst.rows - 1 |
||
| 31 | /// tmp := dst.row[m] |
||
| 32 | /// FOR k := 0 TO (a.colsb / 4) - 1 |
||
| 33 | /// FOR n := 0 TO (dst.colsb / 4) - 1 |
||
| 34 | /// tmp.fp32[n] += FP32(a.row[m].fp16[2*k+0]) * |
||
| 35 | /// FP32(b.row[k].fp16[2*n+0]) |
||
| 36 | /// tmp.fp32[n] += FP32(a.row[m].fp16[2*k+1]) * |
||
| 37 | /// FP32(b.row[k].fp16[2*n+1]) |
||
| 38 | /// ENDFOR |
||
| 39 | /// ENDFOR |
||
| 40 | /// write_row_and_zero(dst, m, tmp, dst.colsb) |
||
| 41 | /// ENDFOR |
||
| 42 | /// zero_upper_rows(dst, dst.rows) |
||
| 43 | /// zero_tileconfig_start() |
||
| 44 | /// \endcode |
||
| 45 | /// |
||
| 46 | /// This intrinsic corresponds to the \c TDPFP16PS instruction. |
||
| 47 | /// |
||
| 48 | /// \param dst |
||
| 49 | /// The destination tile. Max size is 1024 Bytes. |
||
| 50 | /// \param a |
||
| 51 | /// The 1st source tile. Max size is 1024 Bytes. |
||
| 52 | /// \param b |
||
| 53 | /// The 2nd source tile. Max size is 1024 Bytes. |
||
| 54 | #define _tile_dpfp16ps(dst, a, b) \ |
||
| 55 | __builtin_ia32_tdpfp16ps(dst, a, b) |
||
| 56 | |||
| 57 | #endif /* __x86_64__ */ |
||
| 58 | #endif /* __AMX_FP16INTRIN_H */ |