harpoon-scheduler: fix and test transformer state fwd