cpuid: Add AMX FP8 detection (#148)

cpuid: Add AMXFP8 detection Signed-off-by: Oleg Zhurakivskyy <oleg.zhurakivskyy@intel.com>
klauspost · Oct 22, 2024 · 60035f3 · 60035f3
1 parent 92d5326
commit 60035f3
Show file tree

Hide file tree

Showing 3 changed files with 218 additions and 214 deletions.
diff --git a/README.md b/README.md
@@ -281,6 +281,7 @@ Exit Code 1
 | AMXBF16            | Tile computational operations on BFLOAT16 numbers                                                                                                                                  |
 | AMXINT8            | Tile computational operations on 8-bit integers                                                                                                                                    |
 | AMXFP16            | Tile computational operations on FP16 numbers                                                                                                                                      |
+| AMXFP8             | Tile computational operations on FP8 numbers                                                                                                                                      |
 | AMXTILE            | Tile architecture                                                                                                                                                                  |
 | APX_F              | Intel APX                                                                                                                                                                          |
 | AVX                | AVX functions                                                                                                                                                                      |

diff --git a/cpuid.go b/cpuid.go
@@ -81,6 +81,7 @@ const (
 	AMXBF16                              // Tile computational operations on BFLOAT16 numbers
 	AMXFP16                              // Tile computational operations on FP16 numbers
 	AMXINT8                              // Tile computational operations on 8-bit integers
+	AMXFP8                               // Tile computational operations on FP8 numbers
 	AMXTILE                              // Tile architecture
 	APX_F                                // Intel APX
 	AVX                                  // AVX functions
@@ -1306,6 +1307,7 @@ func support() flagSet {
 				fs.setIf(ebx&(1<<31) != 0, AVX512VL)
 				// ecx
 				fs.setIf(ecx&(1<<1) != 0, AVX512VBMI)
+				fs.setIf(ecx&(1<<3) != 0, AMXFP8)
 				fs.setIf(ecx&(1<<6) != 0, AVX512VBMI2)
 				fs.setIf(ecx&(1<<11) != 0, AVX512VNNI)
 				fs.setIf(ecx&(1<<12) != 0, AVX512BITALG)