mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
2f17a21b2b
Reland of #116560 and fixes the issued reported by #116695 Pull Request resolved: https://github.com/pytorch/pytorch/pull/117088 Approved by: https://github.com/albanD
28 lines
681 B
C++
28 lines
681 B
C++
#ifndef THP_SERIALIZATION_INC
|
|
#define THP_SERIALIZATION_INC
|
|
|
|
#include <c10/core/StorageImpl.h>
|
|
#include <c10/util/intrusive_ptr.h>
|
|
template <class io>
|
|
void doRead(io fildes, void* buf, size_t nbytes);
|
|
|
|
template <class io>
|
|
void doWrite(io fildes, void* buf, size_t nbytes);
|
|
|
|
// Note that this takes a mutable storage because it may pass through
|
|
// to at::from_blob.
|
|
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
|