frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add wrappers for synchronous GPUDirect Storage APIs (#130633)

Browse Source 
Based in part on https://github.com/NVIDIA/apex/pull/1774

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130633
Approved by: https://github.com/albanD
This commit is contained in:
Mikayla Gawarecki
2024-07-19 13:36:03 -07:00
committed by PyTorch MergeBot
parent 5c78581fc9
commit 5b5e0698a5
20 changed files with 388 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@ -38,6 +38,9 @@
# USE_CUSPARSELT=0
# disables the cuSPARSELt build
#
# USE_CUFILE=0
# disables the cuFile build
#
# USE_FBGEMM=0
# disables the FBGEMM build
#