e9440acb06e1ab7a9d2f6f888a092b115da3155c,python/dgl/backend/pytorch/tensor.py,,,#,13

Before Change

from ... import kernel as K
from ...function.base import TargetCode

TH_VERSION = LooseVersion(th.__version__)

def data_type_dict():
    return {"float16" : th.float16,
            "float32" : th.float32,

After Change

from ...function.base import TargetCode
from ...base import dgl_warning

if LooseVersion(th.__version__) < LooseVersion("1.2.0"):
    dgl_warning("Detected an old version of PyTorch. Suggest using torch>=1.2.0 "
                "for the best experience.")

def data_type_dict():
    return {"float16" : th.float16,
            "float32" : th.float32,
            "float64" : th.float64,

Instances