pytorch/torch/csrc/cuda/GdsFile.h
Mikayla Gawarecki 018e48c337 [Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489)
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
2024-08-15 17:11:52 +00:00

8 lines
158 B
C

#ifndef THCP_GDSFILE_INC
#define THCP_GDSFILE_INC
#include <torch/csrc/python_headers.h>
void initGdsBindings(PyObject* module);
#endif // THCP_GDSFILE_INC