Skip to content

Commit 60035f3

Browse files
authored
cpuid: Add AMX FP8 detection (#148)
cpuid: Add AMXFP8 detection Signed-off-by: Oleg Zhurakivskyy <oleg.zhurakivskyy@intel.com>
1 parent 92d5326 commit 60035f3

File tree

3 files changed

+218
-214
lines changed

3 files changed

+218
-214
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,7 @@ Exit Code 1
281281
| AMXBF16 | Tile computational operations on BFLOAT16 numbers |
282282
| AMXINT8 | Tile computational operations on 8-bit integers |
283283
| AMXFP16 | Tile computational operations on FP16 numbers |
284+
| AMXFP8 | Tile computational operations on FP8 numbers |
284285
| AMXTILE | Tile architecture |
285286
| APX_F | Intel APX |
286287
| AVX | AVX functions |

cpuid.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ const (
8181
AMXBF16 // Tile computational operations on BFLOAT16 numbers
8282
AMXFP16 // Tile computational operations on FP16 numbers
8383
AMXINT8 // Tile computational operations on 8-bit integers
84+
AMXFP8 // Tile computational operations on FP8 numbers
8485
AMXTILE // Tile architecture
8586
APX_F // Intel APX
8687
AVX // AVX functions
@@ -1306,6 +1307,7 @@ func support() flagSet {
13061307
fs.setIf(ebx&(1<<31) != 0, AVX512VL)
13071308
// ecx
13081309
fs.setIf(ecx&(1<<1) != 0, AVX512VBMI)
1310+
fs.setIf(ecx&(1<<3) != 0, AMXFP8)
13091311
fs.setIf(ecx&(1<<6) != 0, AVX512VBMI2)
13101312
fs.setIf(ecx&(1<<11) != 0, AVX512VNNI)
13111313
fs.setIf(ecx&(1<<12) != 0, AVX512BITALG)

0 commit comments

Comments
 (0)