pytorch / cpuinfo

CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

BSD 2-Clause "Simplified" License

970 stars 308 forks source link

Adding support for detection of AMD Zen4 CPUs #151

Closed naveenthangudu closed 1 year ago

naveenthangudu commented 1 year ago

added following changes

new enum cpuinfo_uarch_zen4 for zen4 CPUs
added CPU model ranges under CPU family 0x19 for identifying AMD Zen4 CPUs

meteorfox commented 1 year ago

Thanks @naveenthangudu! I've tested it on both Bergamo and Genoa processors, and they are now detected as Zen 4, as expected. Previously it would just show 'unknown'

@Maratyszcza PTAL, I've requested AMD to make this contribution. I think we should be ready to merge.