diff --git a/pylops/_torchoperator.py b/pylops/_torchoperator.py index 6249c948..b0e94006 100644 --- a/pylops/_torchoperator.py +++ b/pylops/_torchoperator.py @@ -35,7 +35,7 @@ def forward(ctx, x, forw, adj, device, devicetorch): x = x.cpu().detach().numpy() else: # pass x to cupy using DLPack - x = cp.fromDlpack(to_dlpack(x)) + x = cp.from_dlpack(to_dlpack(x)) # apply forward operator y = ctx.forw(x) @@ -46,7 +46,7 @@ def forward(ctx, x, forw, adj, device, devicetorch): y = torch.from_numpy(y).to(ctx.devicetorch) else: # move y to torch and device - y = from_dlpack(y.toDlpack()) + y = from_dlpack(y) return y @staticmethod @@ -56,7 +56,7 @@ def backward(ctx, y): y = y.cpu().detach().numpy() else: # pass x to cupy using DLPack - y = cp.fromDlpack(to_dlpack(y)) + y = cp.from_dlpack(to_dlpack(y)) # apply adjoint operator x = ctx.adj(y) @@ -65,5 +65,5 @@ def backward(ctx, y): if ctx.device == "cpu": x = torch.from_numpy(x).to(ctx.devicetorch) else: - x = from_dlpack(x.toDlpack()) + x = from_dlpack(x) return x, None, None, None, None, None