Convert from Wave dialect to torch.fx

[ftynse]

Let's have it roughly at the level after index sequence analysis (-water-wave-infer-index-exprs pass). We have some tests and this is the current rotation point.

Feel free to spawn subtasks.