[UserWarning] Fixing the warnings appearing in `test_nn.py::test_grou…

…p_rev_res`. (#7486)
dmlc · Jul 1, 2024 · 65b949f · 65b949f
1 parent 489671c
commit 65b949f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/dgl/nn/pytorch/conv/grouprevres.py b/python/dgl/nn/pytorch/conv/grouprevres.py
@@ -32,7 +32,7 @@ def forward(ctx, fn, fn_inverse, num_inputs, *inputs_and_weights):
             outputs = ctx.fn(*x).detach_()
 
         # clear memory of input node features
-        inputs[1].storage().resize_(0)
+        inputs[1].untyped_storage().resize_(0)
 
         # store for backward pass
         ctx.inputs = [inputs]
@@ -63,10 +63,10 @@ def backward(ctx, *grad_outputs):
                 *((inputs[0], outputs) + inputs[2:])
             )
             # clear memory of outputs
-            outputs.storage().resize_(0)
+            outputs.untyped_storage().resize_(0)
 
             x = inputs[1]
-            x.storage().resize_(int(np.prod(x.size())))
+            x.untyped_storage().resize_(int(np.prod(x.size())))
             x.set_(inputs_inverted)
 
         # compute gradients