[llvm] [mlir] [mlir] Add Normalize pass (PR #162266)

[Jacques Pienaar via llvm-commits]

================
@@ -943,6 +943,17 @@ def ConvertShardToMPIPass : Pass<"convert-shard-to-mpi"> {
   ];
 }
 
+//===----------------------------------------------------------------------===//
+// Normalize
+//===----------------------------------------------------------------------===//
+def Normalize : Pass<"normalize", "ModuleOp"> {
----------------
jpienaar wrote:

I don't quite follow this. I may be missing what you are saying as I think this is true and could work even for functions/doesn't even need to be on top-level ops. But it does need to (correct me if I'm wrong) run sequentially due to rng update ordering.

https://github.com/llvm/llvm-project/pull/162266