Enable AMX using pytorch/cpuinfo cpuinfo_has_x86_amx_int8()

- Updated pytorch/cpuinfo adds AMX detect.
- Remove fp16 hack that inferred AMX from having fp16

PiperOrigin-RevId: 641751226

src/configs/hardware-config.c

@@ -120,14 +120,14 @@ static void init_hardware_config(void) {
     hardware_config.use_x86_avx512vnni = hardware_config.use_x86_avx512skx && cpuinfo_has_x86_avx512vnni();
     hardware_config.use_x86_avx512vnnigfni = hardware_config.use_x86_avx512vnni && cpuinfo_has_x86_gfni();
 #if XNN_ENABLE_AVX512FP16
-    hardware_config.use_x86_avx512fp16 = hardware_config.use_x86_avx512vnnigfni && cpuinfo_has_x86_avx512fp16();
+    hardware_config.use_x86_avx512fp16 = cpuinfo_has_x86_avx512fp16();
 #else
     hardware_config.use_x86_avx512fp16 = 0;
 #endif
 #if XNN_ENABLE_AVX512AMX
     // TODO(fbarchard): Use cpuinfo_has_x86_amx_int8 when available.
     // Infer AMX support from Sapphire Rapids having fp16 and amx.
-    hardware_config.use_x86_avx512amx = hardware_config.use_x86_avx512vnnigfni && cpuinfo_has_x86_avx512fp16();
+    hardware_config.use_x86_avx512amx = hardware_config.use_x86_avx512vnnigfni && cpuinfo_has_x86_amx_int8();
 #if XNN_ARCH_X86_64 && defined(__linux__) && !defined(CHROMIUM)
     if (hardware_config.use_x86_avx512amx) {
       size_t status = xnn_syscall(SYS_arch_prctl, ARCH_REQ_XCOMP_PERM, XFEATURE_XTILEDATA, 0);