2023-04-27 09:56:41 +00:00
|
|
|
#ifndef _IMMINTRIN_H_INCLUDED
|
|
|
|
#error "Never use <avx512vnniintrin.h> directly; include <immintrin.h> instead."
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef __AVX512VNNIINTRIN_H_INCLUDED
|
|
|
|
#define __AVX512VNNIINTRIN_H_INCLUDED
|
|
|
|
|
|
|
|
#if !defined(__AVX512VNNI__)
|
|
|
|
#pragma GCC push_options
|
|
|
|
#pragma GCC target("avx512vnni")
|
|
|
|
#define __DISABLE_AVX512VNNI__
|
|
|
|
#endif /* __AVX512VNNI__ */
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_dpbusd_epi32(__m512i __A, __m512i __B, __m512i __C) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpbusd_v16si((__v16si)__A, (__v16si)__B,
|
|
|
|
(__v16si)__C);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_mask_dpbusd_epi32(__m512i __A, __mmask16 __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpbusd_v16si_mask(
|
|
|
|
(__v16si)__A, (__v16si)__C, (__v16si)__D, (__mmask16)__B);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_maskz_dpbusd_epi32(__mmask16 __A, __m512i __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpbusd_v16si_maskz(
|
|
|
|
(__v16si)__B, (__v16si)__C, (__v16si)__D, (__mmask16)__A);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_dpbusds_epi32(__m512i __A, __m512i __B, __m512i __C) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpbusds_v16si((__v16si)__A, (__v16si)__B,
|
|
|
|
(__v16si)__C);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_mask_dpbusds_epi32(__m512i __A, __mmask16 __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpbusds_v16si_mask(
|
|
|
|
(__v16si)__A, (__v16si)__C, (__v16si)__D, (__mmask16)__B);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_maskz_dpbusds_epi32(__mmask16 __A, __m512i __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpbusds_v16si_maskz(
|
|
|
|
(__v16si)__B, (__v16si)__C, (__v16si)__D, (__mmask16)__A);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_dpwssd_epi32(__m512i __A, __m512i __B, __m512i __C) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpwssd_v16si((__v16si)__A, (__v16si)__B,
|
|
|
|
(__v16si)__C);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_mask_dpwssd_epi32(__m512i __A, __mmask16 __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpwssd_v16si_mask(
|
|
|
|
(__v16si)__A, (__v16si)__C, (__v16si)__D, (__mmask16)__B);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_maskz_dpwssd_epi32(__mmask16 __A, __m512i __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpwssd_v16si_maskz(
|
|
|
|
(__v16si)__B, (__v16si)__C, (__v16si)__D, (__mmask16)__A);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_dpwssds_epi32(__m512i __A, __m512i __B, __m512i __C) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpwssds_v16si((__v16si)__A, (__v16si)__B,
|
|
|
|
(__v16si)__C);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_mask_dpwssds_epi32(__m512i __A, __mmask16 __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpwssds_v16si_mask(
|
|
|
|
(__v16si)__A, (__v16si)__C, (__v16si)__D, (__mmask16)__B);
|
|
|
|
}
|
|
|
|
|
2023-05-16 06:11:47 +00:00
|
|
|
__funline __m512i _mm512_maskz_dpwssds_epi32(__mmask16 __A, __m512i __B,
|
|
|
|
__m512i __C, __m512i __D) {
|
2023-04-27 09:56:41 +00:00
|
|
|
return (__m512i)__builtin_ia32_vpdpwssds_v16si_maskz(
|
|
|
|
(__v16si)__B, (__v16si)__C, (__v16si)__D, (__mmask16)__A);
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef __DISABLE_AVX512VNNI__
|
|
|
|
#undef __DISABLE_AVX512VNNI__
|
|
|
|
#pragma GCC pop_options
|
|
|
|
#endif /* __DISABLE_AVX512VNNI__ */
|
|
|
|
|
|
|
|
#endif /* __AVX512VNNIINTRIN_H_INCLUDED */
|