5.7 Hemi Swap Tech question

It sounds like the ECU isn't getting power while cranking and that's why it won't stay running until you switch the key back to on. Check to make sure it's staying powered up during cranking and then go from there is what I'd do.
-Jon