reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-17 f7c4a3cfd07adede3308f8d9d3d7315427d90a7c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#pragma once
 
// STOP!!! Thinking of including this header directly?  Please
// read Note [TH abstraction violation]
 
#include <THC/THCStorage.h>
// Should work with THStorageClass
#include <TH/THStorageFunctions.hpp>
 
#include <c10/core/ScalarType.h>
 
#include <cuda.h>
#include <cuda_runtime.h>
#include <cuda_fp16.h>
 
THC_API THCStorage* THCStorage_new(THCState* state, caffe2::TypeMeta);
 
THC_API void THCStorage_retain(THCState *state, THCStorage *storage);
 
THC_API void THCStorage_resize(THCState *state, THCStorage *storage, ptrdiff_t size);
THC_API int THCStorage_getDevice(THCState* state, const THCStorage* storage);
 
THC_API THCStorage* THCStorage_newWithDataAndAllocator(
  THCState *state, at::ScalarType scalar_type,
  at::DataPtr&& data, ptrdiff_t size,
  at::Allocator* allocator);