项目文件夹

文件

T

文件历史

nv-dlasalle 1f2e696080 Prevent users from attempting to pin PyTorch non-contiguous tensors or views only encompassing part of tensor. (#3992 )

* Disable pinning non-contiguous memory

* Prevent views from being converted for write

* Fix linting

* Add unit tests

* Improve error message for users

* Switch to pytest function

* exclude mxnet and tensorflow from inplace pinning

* Add skip

* Restrict to pytorch backend

* Use backend to retrieve device

* Fix capitalization in decorator

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

2022-05-16 14:49:36 -07:00

__init__.py

[Bugfix] Bug fixes in new dataloader (#3727 )

2022-02-22 01:30:31 +08:00

checks.py

[Feature] Enable UVA for GPU PinSAGE and RandomWalk (#3857 )

2022-04-11 20:01:23 +08:00

data.py

[Sampling] Cluster-GCN and ShaDow-GNN DataLoader (#3487 )

2021-11-16 13:16:25 +08:00

exception.py

[Sampling] New sampling pipeline plus asynchronous prefetching (#3665 )