pytorch/torch/csrc/serialization.h
Iurii Zdebskyi 3a8d7463bd Enabled BFloat16 storage (#21523)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21523
ghimport-source-id: 698b3cbd6b21c09b9ff8bf8011980df8e35c33b0

Test Plan: Imported from OSS

Differential Revision: D15819368

Pulled By: izdeby

fbshipit-source-id: f6b3bba7b3ca8ee677bd80a231dbb3920c07d61c
2019-07-09 21:51:06 -07:00

26 lines
623 B
C++

#ifndef THP_SERIALIZATION_INC
#define THP_SERIALIZATION_INC
#include <torch/csrc/generic/serialization.h>
#include <TH/THGenerateAllTypes.h>
#include <torch/csrc/generic/serialization.h>
#include <TH/THGenerateHalfType.h>
#include <torch/csrc/generic/serialization.h>
#include <TH/THGenerateBoolType.h>
#include <torch/csrc/generic/serialization.h>
#include <TH/THGenerateBFloat16Type.h>
#include <torch/csrc/generic/serialization.h>
#include <TH/THGenerateQTypes.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);
#endif