search

openxla / xla

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

Unify indexing map computation for reduction emitters. #14282

Closed copybara-service[bot] closed 2 days ago

copybara-service[bot] commented 2 days ago

Unify indexing map computation for reduction emitters.

Currently, we have similar overlapping logic all over the place. This can all be unified in a single function.