Skip to content

Commit

Permalink
Move to test on GPU
Browse files Browse the repository at this point in the history
  • Loading branch information
chaithyagr committed Nov 8, 2024
1 parent 986fb96 commit d4edc58
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/GPU/example_learn_samples.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def __init__(self, inital_trajectory):
data=torch.Tensor(inital_trajectory),
requires_grad=True,
)
self.operator = get_operator("gpunufft", wrt_data=True, wrt_traj=True)(
self.operator = get_operator("cufinufft", wrt_data=True, wrt_traj=True)(
self.trajectory.detach().cpu().numpy(),
shape=(256, 256),
density=True,
Expand Down

0 comments on commit d4edc58

Please sign in to comment.