The computing landscape has shifted from homogeneous x86 architectures to a heterogeneous mix including ARM and RISC-V, introducing challenges for high performance computing. Simultaneously, numerical formats have diversified, with modern deep learning workloads favoring low- and mixed-precision arithmetic to boost efficiency. This complexity demands advanced compiler frameworks and portable performance tools. This work explores using the Exo framework to automatically generate optimized, mixed-integer precision gemm micro-kernels for ARM-NEON and RISC-V Vector, easing the burden of architecture-aware tuning for developers of transformer pipelines. Our experimental evaluation on state-of-the-art, low-power CPUs shows competitive performance for mixed-precision integer arithmetic using a representative quantized BERT model. The benefits can be expected to carry over to convolutional neural networks.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Generation of Mixed-Precision Kernels for Quantized Transformer Encoders with Exo

  • Adrián Castelló,
  • Héctor Martínez,
  • Francisco D. Igual,
  • Enrique S. Quintana-Ortí

摘要

The computing landscape has shifted from homogeneous x86 architectures to a heterogeneous mix including ARM and RISC-V, introducing challenges for high performance computing. Simultaneously, numerical formats have diversified, with modern deep learning workloads favoring low- and mixed-precision arithmetic to boost efficiency. This complexity demands advanced compiler frameworks and portable performance tools. This work explores using the Exo framework to automatically generate optimized, mixed-integer precision gemm micro-kernels for ARM-NEON and RISC-V Vector, easing the burden of architecture-aware tuning for developers of transformer pipelines. Our experimental evaluation on state-of-the-art, low-power CPUs shows competitive performance for mixed-precision integer arithmetic using a representative quantized BERT model. The benefits can be expected to carry over to convolutional neural networks.