Make sure all data is on the correct device for metatomic#107

Luthaf merged 1 commit intomainLuthaf/vesin:mainfrom

fix-metatomicLuthaf/vesin:fix-metatomicCopy head branch name to clipboard