fix deprecation warning (dmlc#1147)

ChenAris · Dec 28, 2019 · 655d756 · 655d756
1 parent 913e324
commit 655d756
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/dgl/backend/pytorch/tensor.py b/python/dgl/backend/pytorch/tensor.py
@@ -333,7 +333,7 @@ def backward(ctx, grad_out):
         reducer, binary_op, graph, lhs, rhs, lhs_map, rhs_map, out_map, \
             lhs_data_nd, rhs_data_nd, feat_shape, degs \
             = ctx.backward_cache
-        out_data, = ctx.saved_variables
+        out_data, = ctx.saved_tensors
         out_data_nd = zerocopy_to_dgl_ndarray(out_data)
         grad_lhs = None
         grad_rhs = None
@@ -395,7 +395,7 @@ def forward(ctx, reducer, graph, target, in_data, out_size, in_map,
     def backward(ctx, grad_out):
         reducer, graph, target, in_map, out_map, in_data_nd, degs \
             = ctx.backward_cache
-        out_data, = ctx.saved_variables
+        out_data, = ctx.saved_tensors
         out_data_nd = zerocopy_to_dgl_ndarray(out_data)
         grad_in = None
         if reducer == 'mean':