ciflow/trunk/195661: [ROCm] Tune MediumRadixSort items_per_thread for small sort sizes

发生了什么

Use 4 items per thread for 1024/2048 element sorts on ROCm, while keeping 8 items per thread for 4096 element sorts. Benchmarks showed speedups on MI350X (gfx950) and MI450 (gfx1250) for the smaller sizes, but a regression at 4096 due to the larger thread block. CUDA behavior is unchanged.

摘要按规则整理自下方来源原文

来源