Abstract:Weight quantization is well adapted to cope with the ever-growing complexity of the deep neural network (DNN) model. Diversified quantization schemes lead to diverse quantized bit width and formats of the weights, thereby, subject to different hardware implementations. Such variety prevents a general NPU to leverage different quantization schemes to gain performance and energy efficiency. More importantly, a trend of quantization diversity emerges that applies multiple quantization schemes to different fine-grained structures (e.g., a layer or a channel of weight) of a DNN. Therefore, a general architecture is desired to exploit varied quantization schemes. The crossbar-based processing-in-memory (PIM) architecture, a promising DNN accelerator, is well known for its highly efficient matrix-vector multiplication. However, PIM suffers from the inflexible intracrossbar data path because the weight is stationary on the crossbar and binds to the “add” operation along the bitline. Therefore, many nonuniform quantization methods must rollback the quantization before mapping the weights onto the crossbar. Counterintuitively, this article discovers a unique opportunity of the PIM architecture to exploit varied quantization schemes. We first transform the quantization diversity problem into a consistency problem by aligning the bit with the same magnitude along the same bitline of the crossbar. Consequently, such naive weight mapping causes many square hollows of idle PIM cells. We then propose a novel spatial mapping to exempt these “hollow” crossbar from the intercrossbar data path. To further squeeze the weights on fewer crossbars, we decouple the intracrossbar data path from the hardware bitline by a novel temporal scheduling, so that bits with different magnitudes can be placed on cells along the same bitline. Finally, the proposed IVQ includes a temporal pipeline to avoid the introduced stalling cycles, and a data flow with delicate control mechanisms for the new intra and intercrossbar data paths. Putting all together, IVQ achieves $19.7\times $ , $10.7\times $ , $4.7\times \sim 63.4\times $ , $91.7\times $ speedup, and $17.7\times $ , $5.1\times $ , $5.7\times \sim 68.1\times $ , $541\times $ energy savings over two PIM accelerators (ISAAC and CASCADE), two customized quantization accelerators (based on ASIC and FPGA), and NVIDIA RTX 2080 GPU, respectively.

Accelerating a Triton Fused Kernel for W4A16 Quantized Inference with SplitK work decomposition

HotaQ: Hardware Oriented Token Adaptive Quantization for Large Language Models

Fast Matrix Multiplications for Lookup Table-Quantized LLMs

A Speed Odyssey for Deployable Quantization of LLMs

TP-Aware Dequantization

COMET: Towards Partical W4A4KV4 LLMs Serving

QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference

FlattenQuant: Breaking Through the Inference Compute-bound for Large Language Models with Per-tensor Quantization

KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization

Fast and Efficient 2-bit LLM Inference on GPU: 2/4/16-bit in a Weight Matrix with Asynchronous Dequantization

ABQ-LLM: Arbitrary-Bit Quantized Inference Acceleration for Large Language Models

Efficient Arbitrary Precision Acceleration for Large Language Models on GPU Tensor Cores

TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding

MixPE: Quantization and Hardware Co-design for Efficient LLM Inference

Accelerating truss decomposition on heterogeneous processors

Dual Grained Quantization: Efficient Fine-Grained Quantization for LLM

IVQ: In-Memory Acceleration of DNN Inference Exploiting Varied Quantization

Fast inference with Kronecker-sparse matrices

Agile-Quant: Activation-Guided Quantization for Faster Inference of LLMs on the Edge

FP6-LLM: Efficiently Serving Large Language Models Through FP6-Centric Algorithm-System Co-Design