Caffe2 - C++ API
A deep learning, cross platform ML framework
Storage.cpp
1 #define __STDC_FORMAT_MACROS
2 
3 #include <torch/csrc/python_headers.h>
4 #ifdef _MSC_VER
5 #include <Windows.h>
6 #endif
7 #include <structmember.h>
8 
9 #define THP_HOST_HALF
10 
11 #include <TH/TH.h>
12 // See Note [TH abstraction violation]
13 // - Used to get at the allocator associated with a storage
14 #include <TH/THStorageFunctions.hpp>
15 #include <libshm.h>
16 #include <torch/csrc/THP.h>
17 #include <torch/csrc/copy_utils.h>
18 #include <torch/csrc/DynamicTypes.h>
19 
20 #include <torch/csrc/generic/Storage.cpp>
21 #include <TH/THGenerateAllTypes.h>
22 
23 #include <torch/csrc/generic/Storage.cpp>
24 #include <TH/THGenerateHalfType.h>
25 
26 #include <torch/csrc/generic/Storage.cpp>
27 #include <TH/THGenerateBoolType.h>
28 
29 template<>
31  if (ptr) {
32  THStorage_free(ptr);
33  }
34 }