move tensor if necessary (like device_map: auto with >1 gpu)

#2
No description provided.
RangiLyu changed pull request status to merged
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment