Add special-case optimization to move Allocs that are inputs to Join to the Gpu.