trunk/df5fda033dc2de824e73fff97bdbd808d80ebed0: [MPS] Use histc and index_add_ atomic kernels for bincount (#201045)
- PyTorch 近 90 天出现 2405 次
- PyTorch 近 90 天第 2373 次发版
- 上一次：同一天稍早 · trunk/4862ce0dc3fd31f37cc1b0c9fa55d49eefa21d0c: [MPS] Add special_airy_ai (#195264)
发生了什么
Fixes #201038 MPS Graph integer scatter add was slow when most inputs fell ine a single bin. This PR changes it so that unweighted bincount uses histc kernel as on CUDA and weighted bincount uses index add. Perf x p weights main ms after ms speedup torch.randint(1, 64, (4194304,)) 0.999 - 6776.655 0.533 12718x torch.randint(1, 630, (4194304,)) 0.999 - 6269.008 0.548 11442x torch.randint(1, 630, (4194304,)) 0.999 tor…
摘要按规则整理自下方来源原文