search

openxla / xla

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

[XLA:GPU] Fix `GpuFloatSupport` for reductions. #17899

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

[XLA:GPU] Fix GpuFloatSupport for reductions.

Presumably, we can lower a reduction computation without normalizing it if all the instructions in the reducer are supported.