dmlc--dgl
Quan (Andy) Gan
3f138ebaaf
* fixes * fix * more fixes * update * oops * lint? * temporarily revert - will fix in another PR * more fixes * skipping mxnet test * address comments * fix DDP * fix edge dataloader exclusion problems * stupid bug * fix * use_uvm option * fix * fixes * fixes * fixes * fixes * add evaluation for cluster gcn and ddp * stupid bug again * fixes * move sanity checks to only support DGLGraphs * pytorch lightning compatibility fixes * remove * poke * more fixes * fix * fix * disable test * docstrings * why is it getting a memory leak? * fix * update * updates and temporarily disable forkingpickler * update * fix? * fix? * oops * oops * fix * lint * huh * uh * update * fix * made it memory efficient * refine exclude interface * fix tutorial * fix tutorial * fix graph duplication in CPU dataloader workers * lint * lint * Revert "lint" This reverts commit 805484dd553695111b5fb37f2125214a6b7276e9. * Revert "lint" This reverts commit 0bce411b2b415c2ab770343949404498436dc8b2. * Revert "fix graph duplication in CPU dataloader workers" This reverts commit 9e3a8cf34c175d3093c773f6bb023b155f2bd27f. Co-authored-by: xiny <xiny@nvidia.com> Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
11 行
327 B
Python
11 行
327 B
Python
"""Feature storage classes for DataLoading"""
|
|
from .. import backend as F
|
|
|
|
from .base import *
|
|
from .numpy import *
|
|
# Defines the name TensorStorage
|
|
if F.get_preferred_backend() == 'pytorch':
|
|
from .pytorch_tensor import PyTorchTensorStorage as TensorStorage
|
|
else:
|
|
from .tensor import BaseTensorStorage as TensorStorage
|