tree 8650e3b257bba26291a388cfca70a74e27612cf3
parent f7a21576f39c79983161f74e33abc449486f761e
author Quinn Dawkins <quinn.dawkins@gmail.com> 1736541232 -0500
committer GitHub <noreply@github.com> 1736541232 -0500
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJngYQwCRC1aQ7uu5UhlAAAcCcQAEJbmAZJwdkAyTtVTmbgQPlC
 FR6FngqH7LpX8vjwblnuHY143GnCWNs1bXmRtlK3fMelXIq1CzWAmMoVRa7dQrrF
 iensLcF8SRuWo6tCJ0QDGauo/5R53LcQ4ARXYhMZmVd0bkbGPKEvV8XIIkLHq8Ob
 8BhYM8haqZ4CKqD5xyoJN6JuKJ6DUhw+s2VDEXZz4tS+6hvEAy6tDY13TwIF7+T6
 qAct7Zc+f2Y4rbMRnmQNqd7e+VlZz+WMIdSRGW+ywktKWt45ytJepgtcWiqRozsK
 QrwB+ElJx/hT3hjJcN3/+pkNZbqxLm71BcAHrabY/idTlG6qOs8PjqwuzQdckCxt
 OGD2gSFAY5j+sN73rH1bmaEgkqk5KUhEonMkYg82EJFQlOrR282KxoVR7TTZtAQx
 iiK2b0T+d/drvKCyR6AZXODFWCc9XPEaqO9uP1o9Tf2mqhXwQNlF64ETJgrTh0pu
 9dIBUVjYMeq0eWRH8WJ8jwvpZOwGcRY+6WU6pQiLCzRvzoSmVM5KtutsuSsNv7ck
 fjwqwn6EHxPQr6oKf1///AH8+wWA1S9jH0ZltrOhMBuaSBQLMpx1kNhpnVuQYK0X
 NUFqYWj1erq1QyUKHFR9wKQPtAmouVjfbFWwS31e3NcsxYd+p2XlFFJqf9pe2R98
 GZApLtI5m2b6QW1cNTzY
 =nP6A
 -----END PGP SIGNATURE-----
 

[LLVMGPU] Use LLVMGPUDistribute for small input scatters (#19670)

If the scattered slice is small then we will end up only distributing
the batch dimensions to workgroups. This causes bufferization to fail in
LLVMGPUTileAndFuse because the workgroup level `extract_slice` will fold
away.

Fixes #19639