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

[Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489)

Browse Source 
Reland #130633

USE_CUFILE turned off by default in this version
Pull Request resolved: https://github.com/pytorch/pytorch/pull/133489
Approved by: https://github.com/albanD
This commit is contained in:
Mikayla Gawarecki
2024-08-15 17:11:52 +00:00
committed by PyTorch MergeBot
parent c23dceb8f1
commit 018e48c337
19 changed files with 391 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
build_variables.bzl
View File

@ -773,6 +773,7 @@ libtorch_python_cuda_core_sources = [
"torch/csrc/cuda/shared/cudart.cpp",
"torch/csrc/cuda/shared/nvtx.cpp",
"torch/csrc/cuda/utils.cpp",
"torch/csrc/cuda/GdsFile.cpp",
]
libtorch_python_cuda_sources = libtorch_python_cuda_core_sources + [