Issue with intermediate tensors not loading to GPU (Fixes #8)

jemisjoky / TorchMPS

PyTorch toolbox for matrix product state models

MIT License

138 stars 31 forks source link

Closed jemisjoky closed 4 years ago

jemisjoky commented 4 years ago

Fixes the issue raised in #8 involving intermediate tensors not loading to a CUDA device when it is available.