Enhancing Transformers: Accelerating Nonlinear Function Computation On RISC-V Vector Processor
summary
Describes a method and custom RISC‑V vector extension instructions for faster exponential function evaluation used inside nonlinearities in transformer models (GeLU, Softmax). Presents an algorithmic approximation with tunable precision parameters plus a hardware microarchitecture implementation (about 95k gates per fp32 exp) and reports speedups versus the standard RISC‑V vector extension (≈4.44× for GeLU, ≈10.7× for Softmax) supporting FP16/BF16/FP32 data types.
AI-written summary · about
related talks
24mMoving to RISC-V Vector: A Practical Journey of AI Operator Optimization
28mAn Efficient Implementation of TensorFlow Lite for RISC-V Vectors
13mUnleashing ML Processing Power Through RISC-V Vectors: Applications, Implementation and...
11mDemo: Running Transformers on Semidynamic's "All-In-One" Vector and Tensor Unit
18mLessons Learned in Using RISC-V for Generative AI and Where We Can Go from Here
10mOptimize Openblas by RISC-V "V" Vector Extension
20mBoosting Video Codec With RISC-V Vector Extension
34mRISC V Vector Extension Proposal
