openxla / xla

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

[XLA:GPU] Enable new mlir loop emitter by default. #14235

Closed copybara-service[bot] closed 5 days ago

copybara-service[bot] commented 5 days ago

[XLA:GPU] Enable new mlir loop emitter by default.

It can be disabled again by setting --xla_gpu_mlir_emitter_level=0

Remove two unrolling tests. With MLIR emitters enabled, we do unroll in these two cases, but benchmarking shows that it is at least as fast as with the old emitters.

Remove one test in element_wise_row_vectorization.hlo. It checks that we do not vectorize for multi-output fusions, but there is no reason why we shouldn't. With MLIR emitters, we would vectorize here.

Reverts e7360d66eb2f2e928d85e1c3c9f7717d701082f2