add Hygon Dhyana CPU Vendor ID support (#29)

Signed-off-by: hygonsoc <hygonsoc@gmail.com>

Author: hygonsoc

README.md

@@ -84,6 +84,7 @@ Package home: https://github.com/klauspost/cpuid
 * **VMware**
 * **XenHVM**
 * **Bhyve**
+* **Hygon**
 
 # installing
 

cpuid.go

@@ -27,6 +27,7 @@ const (
 	VMware
 	XenHVM
 	Bhyve
+	Hygon
 )
 
 const (
@@ -473,6 +474,11 @@ func (c CPUInfo) AMD() bool {
 	return c.VendorID == AMD
 }
 
+// Hygon returns true if vendor is recognized as Hygon
+func (c CPUInfo) Hygon() bool {
+	return c.VendorID == Hygon
+}
+
 // Transmeta returns true if vendor is recognized as Transmeta
 func (c CPUInfo) Transmeta() bool {
 	return c.VendorID == Transmeta
@@ -626,7 +632,7 @@ func logicalCores() int {
 		}
 		_, b, _, _ := cpuidex(0xb, 1)
 		return int(b & 0xffff)
-	case AMD:
+	case AMD, Hygon:
 		_, b, _, _ := cpuid(1)
 		return int((b >> 16) & 0xff)
 	default:
@@ -648,7 +654,7 @@ func physicalCores() int {
 	switch vendorID() {
 	case Intel:
 		return logicalCores() / threadsPerCore()
-	case AMD:
+	case AMD, Hygon:
 		if maxExtendedFunction() >= 0x80000008 {
 			_, _, c, _ := cpuid(0x80000008)
 			return int(c&0xff) + 1
@@ -672,6 +678,7 @@ var vendorMapping = map[string]Vendor{
 	"VMwareVMware": VMware,
 	"XenVMMXenVMM": XenHVM,
 	"bhyve bhyve ": Bhyve,
+	"HygonGenuine": Hygon,
 }
 
 func vendorID() Vendor {
@@ -744,7 +751,7 @@ func (c *CPUInfo) cacheSize() {
 				c.Cache.L3 = size
 			}
 		}
-	case AMD:
+	case AMD, Hygon:
 		// Untested.
 		if maxExtendedFunction() < 0x80000005 {
 			return

cpuid_test.go

@@ -599,6 +599,16 @@ func TestAMD(t *testing.T) {
 	t.Log("TestAMD:", got)
 }
 
+// Hygon returns true if vendor is recognized as Hygon
+func TestHygon(t *testing.T) {
+	got := CPU.Hygon()
+	expected := CPU.VendorID == Hygon
+	if got != expected {
+		t.Fatalf("TestHygon: expected %v, got %v", expected, got)
+	}
+	t.Log("TestHygon:", got)
+}
+
 // Transmeta returns true if vendor is recognized as Transmeta
 func TestTransmeta(t *testing.T) {
 	got := CPU.Transmeta()

private/cpuid.go

@@ -20,6 +20,7 @@ const (
 	msvm // Microsoft Hyper-V or Windows Virtual PC
 	vmware
 	xenhvm
+	hygon
 )
 
 const (
@@ -466,6 +467,11 @@ func (c cpuInfo) amd() bool {
 	return c.vendorid == amd
 }
 
+// Hygon returns true if vendor is recognized as Hygon
+func (c cpuInfo) hygon() bool {
+	return c.vendorid == hygon
+}
+
 // Transmeta returns true if vendor is recognized as Transmeta
 func (c cpuInfo) transmeta() bool {
 	return c.vendorid == transmeta
@@ -619,7 +625,7 @@ func logicalCores() int {
 		}
 		_, b, _, _ := cpuidex(0xb, 1)
 		return int(b & 0xffff)
-	case amd:
+	case amd, hygon:
 		_, b, _, _ := cpuid(1)
 		return int((b >> 16) & 0xff)
 	default:
@@ -641,7 +647,7 @@ func physicalCores() int {
 	switch vendorID() {
 	case intel:
 		return logicalCores() / threadsPerCore()
-	case amd:
+	case amd, hygon:
 		if maxExtendedFunction() >= 0x80000008 {
 			_, _, c, _ := cpuid(0x80000008)
 			return int(c&0xff) + 1
@@ -664,6 +670,7 @@ var vendorMapping = map[string]vendor{
 	"Microsoft Hv": msvm,
 	"VMwareVMware": vmware,
 	"XenVMMXenVMM": xenhvm,
+	"HygonGenuine": hygon,
 }
 
 func vendorID() vendor {
@@ -736,7 +743,7 @@ func (c *cpuInfo) cacheSize() {
 				c.cache.l3 = size
 			}
 		}
-	case amd:
+	case amd, hygon:
 		// Untested.
 		if maxExtendedFunction() < 0x80000005 {
 			return

private/cpuid_test.go

@@ -601,6 +601,16 @@ func TestAMD(t *testing.T) {
 	t.Log("TestAMD:", got)
 }
 
+// Hygon returns true if vendor is recognized as Hygon
+func TestHygon(t *testing.T) {
+	got := cpu.hygon()
+	expected := cpu.vendorid == hygon
+	if got != expected {
+		t.Fatalf("TestHygon: expected %v, got %v", expected, got)
+	}
+	t.Log("TestHygon:", got)
+}
+
 // Transmeta returns true if vendor is recognized as Transmeta
 func TestTransmeta(t *testing.T) {
 	got := cpu.transmeta()