[MLIR][SCF] Fix normalizeForallOp helper function (#138615)
Previously the `normalizeForallOp` function did not work properly, since
the newly created op was not being returned in addition to the op
failing verification.
This patch fixes the helper function and adds a unit test for it.