后端最佳实践

x86 CPU

现代 x86 CPU 上的编译工作负载通常通过单指令多数据 (SIMD) 指令集进行优化。SIMD 是一种典型的并行处理技术，适用于高性能计算，例如深度学习模型训练和推理。应用 SIMD 后，每个计算单元在任何给定的时间段内都执行相同的指令，但使用不同的分配数据。最常部署的 x86 指令集架构 (ISA) 支持 SIMD，包括 AVX、AVX2、AVX-512 和 AMX。

您可以使用 collect_env 脚本 检查机器支持的 ISA。由于该脚本提供了 PyTorch 的完整环境信息，因此我们可以使用 grep 来提取包含 ISA 信息的行

python collect_env.py | grep "a[(v|m)]x"

通常，如果支持 AVX-512，则应观察到以“avx512”开头的指令（如 avx512f、avx512bw、avx512_vnni）。如果支持 AMX，则应观察到以“amx”开头的指令（如 amx_tile、amx_bf16、amx_int8）。

具体而言，对于启用了 AMX 指令的服务器，可以通过 利用 AMX 进一步提高工作负载性能。