Merge pull request #32 from mythi/vnni

Add AVX512-VNNI detection

cpuid.go

@@ -73,6 +73,7 @@ const (
 	AVX512BW                // AVX-512 Byte and Word Instructions
 	AVX512VL                // AVX-512 Vector Length Extensions
 	AVX512VBMI              // AVX-512 Vector Bit Manipulation Instructions
+	AVX512VNNI              // AVX-512 Vector Neural Network Instructions
 	MPX                     // Intel MPX (Memory Protection Extensions)
 	ERMS                    // Enhanced REP MOVSB/STOSB
 	RDTSCP                  // RDTSCP Instruction
@@ -131,6 +132,7 @@ var flagNames = map[Flags]string{
 	AVX512BW:    "AVX512BW",    // AVX-512 Byte and Word Instructions
 	AVX512VL:    "AVX512VL",    // AVX-512 Vector Length Extensions
 	AVX512VBMI:  "AVX512VBMI",  // AVX-512 Vector Bit Manipulation Instructions
+	AVX512VNNI:  "AVX512VNNI",  // AVX-512 Vector Neural Network Instructions
 	MPX:         "MPX",         // Intel MPX (Memory Protection Extensions)
 	ERMS:        "ERMS",        // Enhanced REP MOVSB/STOSB
 	RDTSCP:      "RDTSCP",      // RDTSCP Instruction
@@ -440,6 +442,11 @@ func (c CPUInfo) AVX512VBMI() bool {
 	return c.Features&AVX512VBMI != 0
 }
 
+// AVX512VNNI indicates support of AVX-512 Vector Neural Network Instructions
+func (c CPUInfo) AVX512VNNI() bool {
+	return c.Features&AVX512VNNI != 0
+}
+
 // MPX indicates support of Intel MPX (Memory Protection Extensions)
 func (c CPUInfo) MPX() bool {
 	return c.Features&MPX != 0
@@ -955,6 +962,9 @@ func support() Flags {
 				if ecx&(1<<1) != 0 {
 					rval |= AVX512VBMI
 				}
+				if ecx&(1<<11) != 0 {
+					rval |= AVX512VNNI
+				}
 			}
 		}
 	}

cpuid_test.go

@@ -554,7 +554,7 @@ func TestAVX512VL(t *testing.T) {
 	t.Log("AVX512VL Support:", got)
 }
 
-// TestAVX512VL tests AVX512VBMI() function (AVX-512 Vector Bit Manipulation Instructions)
+// TestAVX512VBMI tests AVX512VBMI() function (AVX-512 Vector Bit Manipulation Instructions)
 func TestAVX512VBMI(t *testing.T) {
 	got := CPU.AVX512VBMI()
 	expected := CPU.Features&AVX512VBMI == AVX512VBMI
@@ -564,6 +564,16 @@ func TestAVX512VBMI(t *testing.T) {
 	t.Log("AVX512VBMI Support:", got)
 }
 
+// TestAVX512VNNI tests AVX512VNNI() function (AVX-512 Vector Neural Network Instructions)
+func TestAVX512VNNI(t *testing.T) {
+	got := CPU.AVX512VNNI()
+	expected := CPU.Features&AVX512VNNI == AVX512VNNI
+	if got != expected {
+		t.Fatalf("AVX512VNNI: expected %v, got %v", expected, got)
+	}
+	t.Log("AVX512VNNI Support:", got)
+}
+
 // TestMPX tests MPX() function (Intel MPX (Memory Protection Extensions))
 func TestMPX(t *testing.T) {
 	got := CPU.MPX()