New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fixed device mismatch in energy_spectrum computation #108

Merged

danielkelshaw merged 1 commit into main from elise-bugfix/torch_energy_spectrum

Sep 4, 2023

Contributor

eliseoe commented Sep 4, 2023

I addressed the device mismatch issue by specifying the device type in the torch.zeros call within the energy spectrum computation, see below:

# Before the change
torch.zeros((tuple([*leading_dims]) + tuple([torch.max(intk) + 1])))

# After the change
torch.zeros((tuple([*leading_dims]) + tuple([torch.max(intk) + 1])), device=self.device)


          Fixed device mismatch in energy_spectrum computation

eecc0af

eliseoe requested a review from danielkelshaw

September 4, 2023 11:23

eliseoe linked an issue

that may be closed by this pull request

Correct pytorch energy spectrum device assignment #107

Closed

danielkelshaw merged commit 2338b25 into main

1 check failed

danielkelshaw deleted the elise-bugfix/torch_energy_spectrum branch

September 4, 2023 12:44

Member

danielkelshaw commented Sep 4, 2023

That's great, thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment