[MLIR] Generic checkpointing implementation

[Pangoraw]

create generic checkpointing implementations for

• affine.for
• scf.for
• stablhlo.while in for loop mode