search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.39k stars 356 forks source link

Clean up tile size computation for row reductions. #14284

Closed copybara-service[bot] closed 2 days ago

copybara-service[bot] commented 2 days ago

Clean up tile size computation for row reductions.

Also don't attempt to peel the vectorized dimension.