mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
30fb2c4aba
Let's have some fun. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78828 Approved by: https://github.com/ezyang
24 lines
520 B
C++
24 lines
520 B
C++
#ifndef THP_SERIALIZATION_INC
|
|
#define THP_SERIALIZATION_INC
|
|
|
|
template <class io>
|
|
void doRead(io fildes, void* buf, size_t nbytes);
|
|
|
|
template <class io>
|
|
void doWrite(io fildes, void* buf, size_t nbytes);
|
|
|
|
template <class io>
|
|
void THPStorage_writeFileRaw(
|
|
c10::StorageImpl* self,
|
|
io fd,
|
|
bool save_size,
|
|
uint64_t element_size);
|
|
|
|
template <class io>
|
|
c10::intrusive_ptr<c10::StorageImpl> THPStorage_readFileRaw(
|
|
io fd,
|
|
c10::intrusive_ptr<c10::StorageImpl> storage,
|
|
uint64_t element_size);
|
|
|
|
#endif
|