Skip to content

Commit ecad2fd

Browse files
committed
feat: add support got Graviton 4 cpu detection
First step to test m8g is to identify if you run on m8g. Signed-off-by: Egor Lazarchuk <[email protected]>
1 parent 9ff2c7b commit ecad2fd

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/framework/utils_cpuid.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ class CpuModel(str, Enum):
2828
AMD_GENOA = "AMD_GENOA"
2929
ARM_NEOVERSE_N1 = "ARM_NEOVERSE_N1"
3030
ARM_NEOVERSE_V1 = "ARM_NEOVERSE_V1"
31+
ARM_NEOVERSE_V2 = "ARM_NEOVERSE_V2"
3132
INTEL_SKYLAKE = "INTEL_SKYLAKE"
3233
INTEL_CASCADELAKE = "INTEL_CASCADELAKE"
3334
INTEL_ICELAKE = "INTEL_ICELAKE"
@@ -41,7 +42,11 @@ class CpuModel(str, Enum):
4142
"Intel(R) Xeon(R) Platinum 8375C CPU": "INTEL_ICELAKE",
4243
},
4344
CpuVendor.AMD: {"AMD EPYC 7R13": "AMD_MILAN", "AMD EPYC 9R14": "AMD_GENOA"},
44-
CpuVendor.ARM: {"0xd0c": "ARM_NEOVERSE_N1", "0xd40": "ARM_NEOVERSE_V1"},
45+
CpuVendor.ARM: {
46+
"0xd0c": "ARM_NEOVERSE_N1",
47+
"0xd40": "ARM_NEOVERSE_V1",
48+
"0xd4f": "ARM_NEOVERSE_V2",
49+
},
4550
}
4651

4752

0 commit comments

Comments
 (0)