clang/lib/Headers/amxfp16intrin.h

   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 */