PTX: Fix some lingering issues with stack allocation