implement MFSPR the same way as fu/spr/main_stage.py