amxintrin.h

Go to the documentation of this file.

/*===--------------- amxintrin.h - AMX intrinsics -*- C/C++ -*---------------===
 *
 * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 * See https://llvm.org/LICENSE.txt for license information.
 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 *
 *===------------------------------------------------------------------------===
 */
 
#ifndef __IMMINTRIN_H
#error "Never use <amxintrin.h> directly; include <immintrin.h> instead."
#endif /* __IMMINTRIN_H */
 
#ifndef __AMXINTRIN_H
#define __AMXINTRIN_H
#ifdef __x86_64__
 
#define __DEFAULT_FN_ATTRS \
  __attribute__((__always_inline__, __nodebug__,  __target__("amx-tile")))
 
static __inline__ void __DEFAULT_FN_ATTRS
_tile_loadconfig(const void *__config)
{
  __builtin_ia32_tile_loadconfig(__config);
}
 
static __inline__ void __DEFAULT_FN_ATTRS
_tile_storeconfig(void *__config)
{
  __builtin_ia32_tile_storeconfig(__config);
}
 
static __inline__ void __DEFAULT_FN_ATTRS
_tile_release(void)
{
  __builtin_ia32_tilerelease();
}
 
#define _tile_loadd(dst, base, stride) \
  __builtin_ia32_tileloadd64((dst), ((const void *)(base)), (__SIZE_TYPE__)(stride))
 
#define _tile_stream_loadd(dst, base, stride) \
  __builtin_ia32_tileloaddt164((dst), ((const void *)(base)), (__SIZE_TYPE__)(stride))
 
#define _tile_stored(dst, base, stride) \
  __builtin_ia32_tilestored64((dst), ((void *)(base)), (__SIZE_TYPE__)(stride))
 
#define _tile_zero(tile) __builtin_ia32_tilezero((tile))
 
#define _tile_dpbssd(dst, src0, src1) __builtin_ia32_tdpbssd((dst), (src0), (src1))
 
#define _tile_dpbsud(dst, src0, src1) __builtin_ia32_tdpbsud((dst), (src0), (src1))
 
#define _tile_dpbusd(dst, src0, src1) __builtin_ia32_tdpbusd((dst), (src0), (src1))
 
#define _tile_dpbuud(dst, src0, src1) __builtin_ia32_tdpbuud((dst), (src0), (src1))
 
#define _tile_dpbf16ps(dst, src0, src1) \
  __builtin_ia32_tdpbf16ps((dst), (src0), (src1))
 
#undef __DEFAULT_FN_ATTRS
 
#endif /* __x86_64__ */
#endif /* __AMXINTRIN_H */