Skip to content

Commit 084028f

Browse files
committed
add detection of AMX_COMPLEX
1 parent a33007c commit 084028f

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

sample/test_util.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@ void putCPUinfo(bool onlyCpuidFeature)
114114
{ Cpu::tAMX_MOVRS, "amx_movrs" },
115115
{ Cpu::tMOVRS, "movrs" },
116116
{ Cpu::tHYBRID, "hybrid" },
117+
{ Cpu::tAMX_COMPLEX, "amx_complex" },
117118
};
118119
for (size_t i = 0; i < NUM_OF_ARRAY(tbl); i++) {
119120
if (cpu.has(tbl[i].type)) printf(" %s", tbl[i].str);

xbyak/xbyak_util.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -589,6 +589,7 @@ class Cpu {
589589
XBYAK_DEFINE_TYPE(95, tAMX_FP8);
590590
XBYAK_DEFINE_TYPE(96, tMOVRS);
591591
XBYAK_DEFINE_TYPE(97, tHYBRID);
592+
XBYAK_DEFINE_TYPE(98, tAMX_COMPLEX);
592593

593594
#undef XBYAK_SPLIT_ID
594595
#undef XBYAK_DEFINE_TYPE
@@ -742,6 +743,7 @@ class Cpu {
742743
if (eax & (1U << 31)) type_ |= tMOVRS;
743744
if (edx & (1U << 4)) type_ |= tAVX_VNNI_INT8;
744745
if (edx & (1U << 5)) type_ |= tAVX_NE_CONVERT;
746+
if (edx & (1U << 8)) type_ |= tAMX_COMPLEX;
745747
if (edx & (1U << 10)) type_ |= tAVX_VNNI_INT16;
746748
if (edx & (1U << 14)) type_ |= tPREFETCHITI;
747749
if (edx & (1U << 19)) type_ |= tAVX10;

0 commit comments

Comments
 (0)