68a978d4de27b6625870e1ea4242075e990922f4,examples/pytorch/jtnn/jtnn/nnutils.py,,cuda,#Any#,6

Before Change

def cuda(tensor):
    if torch.cuda.is_available() and not os.getenv("NOCUDA", None):
        return tensor.cuda()
    else:
        return tensor

After Change

def cuda(x):
    if torch.cuda.is_available() and not os.getenv("NOCUDA", None):
        return x.to(torch.device("cuda"))   // works for both DGLGraph and tensor
    else:
        return tensor

Instances