Fine-Grained Vectorized Merge Sorting on RISC-V: From Register to Cache

TL;DR

This paper introduces a novel, fine-grained vectorized merge sort algorithm for RISC-V architectures, optimizing register operations and cache utilization to improve sorting efficiency.

Contribution

It develops a cache-aware, vectorized merge sort with innovative register transpose and merge schemes tailored for RISC-V's vector capabilities.

Findings

Achieves improved sorting performance on RISC-V hardware.

Introduces efficient register transpose using data proxy techniques.

Designs an asymmetric merging network for better cache utilization.

Abstract

Merge sort as a divide-sort-merge paradigm has been widely applied in computer science fields. As modern reduced instruction set computing architectures like the fifth generation (RISC-V) regard multiple registers as a vector register group for wide instruction parallelism, optimizing merge sort with this vectorized property is becoming increasingly common. In this paper, we overhaul the divide-sort-merge paradigm, from its register-level sort to the cache-aware merge, to develop a fine-grained RISC-V vectorized merge sort (RVMS). From the register-level view, the inline vectorized transpose instruction is missed in RISC-V, so implementing it efficiently is non-trivial. Besides, the vectorized comparisons do not always work well in the merging networks. Both issues primarily stem from the expensive data shuffle instruction. To bypass it, RVMS strides to take register data as the proxy of data shuffle to accelerate the transpose operation, and meanwhile replaces vectorized comparisons with scalar cousin for more light real value swap. On the other hand, as cache-aware merge makes larger data merge in the cache, most merge schemes have two drawbacks: the in-cache merge usually has low cache utilization, while the out-of-cache merging network remains an ineffectively symmetric structure. To this end, we propose the half-merge scheme to employ the auxiliary space of in-place merge to halve the footprint of naive merge sort, and meanwhile copy one sequence to this space to avoid the former data exchange. Furthermore, an asymmetric merging network is developed to adapt to two different input sizes.