fix(autograd): fix different dtype backward bug (#8309)

Oneflow-Inc · May 26, 2022 · dd83698 · dd83698
1 parent d8c57af
commit dd83698
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/python/oneflow/test/modules/test_autograd.py b/python/oneflow/test/modules/test_autograd.py
@@ -111,9 +111,7 @@ def test_scalar_leaf_tensor_backward(test_case):
 
     @autotest(n=1, auto_backward=False, check_graph=False)
     def test_out_grad_with_different_dtype(test_case):
-        device = random_device()
         x = random_tensor(ndim=2, requires_grad=True)
-        x = x.to(device)
         y = x.sum()
         y.backward(torch.tensor(False))
         return x.grad