3 ms·
intel/x86-simd-sort v6.0: https://github.com/intel/x86-simd-sort/releases/tag/v6.0 https://github.com/intel/x86-simd-sort/releases/tag/v6.0 : > Release 6.0 in
by westurner 2y ago
intel/x86-simd-sort v6.0:
https://github.com/intel/x86-simd-sort/releases/tag/v6.0 https://github.com/intel/x86-simd-sort/releases/tag/v6.0 :
> Release 6.0 introduces several key enhancements to our software. The update brings new APIs for partial sorting of key-value pairs, along with comprehensive AVX2 support across all routines. Additionally, PyTorch now leverages AVX-512 and AVX2 key-value sort routines to boost the performance of `torch.sort` and `torch.argsort` functions by up-to 10x (see pytorch/pytorch#127936).
"Adds support for accelerated sorting with x86-simd-sort"
https://github.com/pytorch/pytorch/pull/127936 https://github.com/pytorch/pytorch/pull/127936