Reduce stack usage of HLO->MLIR conversion functions.

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.4k stars 361 forks source link

Closed copybara-service[bot] closed 3 days ago

copybara-service[bot] commented 3 days ago

Reduce stack usage of HLO->MLIR conversion functions.

Don't recurse for elementwise ops.
Set sizes of SmallVectors to the common case (1 for operands, 3 for indices) instead of using the default 6.
Don't materialize input indices of elementwise ops.
Use ValueRange instead of SmallVector where possible.

Also split up SubgraphToMlir.