79b0a50afc
[Unittest][Fix] Several unit tests fixes for Ampere+ and PyTorch 1.12+ ( #4213 )
...
* Fix test_csrmm for tensor core
* unset allow tf32 flag
* update test unified tensor
* skip fp16 for CPU
2022-07-14 19:59:57 +08:00
c793593577
[Feature] Add multi-GPU UnifiedTensor unit test ( #3184 )
...
* Add pytorch-direct version
* remove
* Add multi-gpu unified tensor test for pytorch
* relocate verification step to each process
* reduce number of workers
* add parameter
Co-authored-by: shhssdm <shhssdm@gmail.com >
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-08-02 12:41:03 +08:00
905c0aa578
[Feature][Performance][GPU] Introducing UnifiedTensor for efficient zero-copy host memory access from GPU ( #3086 )
...
* Add pytorch-direct version
* Initial commit of unified tensor
* Merge branch 'master' of https://github.com/davidmin7/dgl
* Remove unnecessary things
* Fix error message
* Fix/Add descriptions
* whitespace fix
* add unpin
* disable IndexSelectCPUFromGPU with no CUDA
* add a newline for unified_tensor.py
* Apply changes based on feedback
* add 'os' module
* skip unified tensor unit test for cpu only
* Update tests/pytorch/test_unified_tensor.py
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
* reflect feedback
Co-authored-by: shhssdm <shhssdm@gmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
2021-07-17 00:06:20 +08:00
Xin Yao