Add reduction group support.
Add reduction group support.

After this, all legacy fusions are supported by MLIR emitters. Currently, there
is still one failure in the JAX test suite, but it is not a reduction fusion.
