Cleanup detection of ARM BF16 extension

pytorch / cpuinfo

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

BSD 2-Clause "Simplified" License

992 stars 311 forks source link

Closed Maratyszcza closed 2 years ago

Maratyszcza commented 2 years ago

Remove unneeded svebf16 extension flag, instead use the Linux feature flag to detect bf16 extension
Group "fp16 arith" and "bf16" together with other floating-point extensions
Rename cpuinfo_has_arm_svebf16 to cpuinfo_has_arm_sve_bf16
Add cpuinfo_has_arm_neon_bf16 API