xref: /aosp_15_r20/external/pytorch/torch/csrc/serialization.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

#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