File size: 836 Bytes
c1af2fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
#pragma once
#include <cstdint>
#include <c10/macros/Export.h>
namespace at::cpu {
TORCH_API bool is_avx2_supported();
TORCH_API bool is_avx512_supported();
// Detect if CPU support Vector Neural Network Instruction.
TORCH_API bool is_avx512_vnni_supported();
// Detect if CPU supports AVX512_BF16 ISA
TORCH_API bool is_avx512_bf16_supported();
// Detect if CPU support Advanced Matrix Extension.
TORCH_API bool is_amx_tile_supported();
// Detect if CPU support Advanced Matrix Extension for fp16.
TORCH_API bool is_amx_fp16_supported();
// Enable the system to use AMX instructions.
TORCH_API bool init_amx();
// Get the L1 cache size per core in Byte
TORCH_API uint32_t L1d_cache_size();
// Get the L2 cache size per core in Byte
TORCH_API uint32_t L2_cache_size();
} // namespace at::cpu
|