Bring MPS from GPU back to CPU

I have some MPS calculations that I want to perform on the GPU. Afterwards, I want to do some final manipulations locally on my CPU that do not require the “heavy” computational horsepower of the GPU.

I am wondering how I can convert an MPS from the device (GPU) back to the CPU? In the examples here I can see that to create it on the GPU I do device(MPS(...)) but what about bringing it back to CPU?

If I try to do my final calculations without bringing it back to the CPU I get this error ERROR: LoadError: Scalar indexing is disallowed.

Any suggestions? Apologies if I missed something obvious here.

You can use the function NDTensors.cpu.

1 Like