We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f6f147a commit 84d5c89Copy full SHA for 84d5c89
torch_sparse/spspmm.py
@@ -37,7 +37,7 @@ def forward(ctx, indexA, valueA, indexB, valueB, m, k, n):
37
@staticmethod
38
def backward(ctx, grad_indexC, grad_valueC):
39
m, k, n = ctx.m, ctx.k, ctx.n
40
- indexA, valueA, indexB, valueB, indexC = ctx.saved_variables
+ indexA, valueA, indexB, valueB, indexC = ctx.saved_tensors
41
42
grad_valueA = grad_valueB = None
43
0 commit comments