yuzu/xbyak
1
0
Fork 0
mirror of https://github.com/herumi/xbyak synced 2024-11-20 16:06:14 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

add tAVX_IFMA

Browse source
This commit is contained in:
MITSUNARI Shigeo 2022-10-06 09:35:51 +09:00
parent cd36e31ea6
commit 95752ebd79
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
sample/test_util.cpp
View file

@ -92,6 +92,7 @@ void putCPUinfo(bool onlyCpuidFeature)
{ Cpu::tAMX_FP16, "amx_fp16" },
{ Cpu::tAVX_VNNI_INT8, "avx_vnni_int8" },
{ Cpu::tAVX_NE_CONVERT, "avx_ne_convert" },
{ Cpu::tAVX_IFMA, "avx_ifma" },
};
for (size_t i = 0; i < NUM_OF_ARRAY(tbl); i++) {
if (cpu.has(tbl[i].type)) printf(" %s", tbl[i].str);

2
xbyak/xbyak_util.h
View file

@ -413,6 +413,7 @@ public:
XBYAK_DEFINE_TYPE(68, tAMX_FP16);
XBYAK_DEFINE_TYPE(69, tAVX_VNNI_INT8);
XBYAK_DEFINE_TYPE(70, tAVX_NE_CONVERT);
XBYAK_DEFINE_TYPE(71, tAVX_IFMA);
#undef XBYAK_SPLIT_ID
#undef XBYAK_DEFINE_TYPE
@ -557,6 +558,7 @@ public:
if (EAX & (1U << 5)) type_ |= tAVX512_BF16;
}
if (EAX & (1U << 21)) type_ |= tAMX_FP16;
if (EAX & (1U << 23)) type_ |= tAVX_IFMA;
if (EDX & (1U << 4)) type_ |= tAVX_VNNI_INT8;
if (EDX & (1U << 5)) type_ |= tAVX_NE_CONVERT;
}