mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
d408324350
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15316 This starts cleaning up the files in c10 according to the module structure we decided on. Move to c10/util: - Half.h, Half-inl.h, Half.cpp, bitcasts.h Move to c10/core: - Device.h, Device.cpp - DeviceType.h, DeviceType.cpp i-am-not-moving-c2-to-c10 Reviewed By: dzhulgakov Differential Revision: D13498493 fbshipit-source-id: dfcf1c490474a12ab950c72ca686b8ad86428f63
34 lines
733 B
C++
34 lines
733 B
C++
#ifndef CAFFE2_CORE_STORAGE_H_
|
|
#define CAFFE2_CORE_STORAGE_H_
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <type_traits>
|
|
#include <typeinfo>
|
|
#include <vector>
|
|
|
|
#include "caffe2/core/allocator.h"
|
|
#include "caffe2/core/common.h"
|
|
#include "caffe2/core/context.h"
|
|
#include "caffe2/core/flags.h"
|
|
#include "caffe2/core/logging.h"
|
|
#include <c10/util/typeid.h>
|
|
|
|
#include <c10/core/Allocator.h>
|
|
#include <c10/core/Device.h>
|
|
#include <c10/core/DeviceType.h>
|
|
#include <c10/util/intrusive_ptr.h>
|
|
#include <c10/core/Storage.h>
|
|
#include <c10/core/StorageImpl.h>
|
|
|
|
namespace caffe2 {
|
|
|
|
using StorageImpl = at::StorageImpl;
|
|
using Storage = at::Storage;
|
|
|
|
} // namespace caffe2
|
|
|
|
#endif // CAFFE2_CORE_STORAGE_H_
|