reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-10 c3765bd24fe73747688a0ec2a550f219c9acb384
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#ifndef THCP_UTILS_H
#define THCP_UTILS_H
 
#define THCPUtils_(NAME) TH_CONCAT_4(THCP,Real,Utils_,NAME)
 
#define THCStoragePtr  TH_CONCAT_3(THC,Real,StoragePtr)
#define THCTensorPtr   TH_CONCAT_3(THC,Real,TensorPtr)
#define THCPStoragePtr TH_CONCAT_3(THCP,Real,StoragePtr)
#define THCPTensorPtr  TH_CONCAT_3(THCP,Real,TensorPtr)
 
#define THCSTensorPtr  TH_CONCAT_3(THCS,Real,TensorPtr)
#define THCSPTensorPtr TH_CONCAT_3(THCSP,Real,TensorPtr)
 
#include <torch/csrc/cuda/override_macros.h>
 
#define THC_GENERIC_FILE "torch/csrc/generic/utils.h"
#include <THC/THCGenerateAllTypes.h>
 
#define THC_GENERIC_FILE "torch/csrc/generic/utils.h"
#include <THC/THCGenerateBoolType.h>
 
#define THC_GENERIC_FILE "torch/csrc/generic/utils.h"
#include <THC/THCGenerateBFloat16Type.h>
#endif