reid from https://github.com/michuanhaohao/reid-strong-baseline
554325746@qq.com
2020-02-05 7c6a958f00e327a8991f48093456de7cb84cf918
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "THC/generic/THCTensorMathMagma.h"
#else
 
#if defined(THC_REAL_IS_FLOAT) || defined(THC_REAL_IS_DOUBLE)
 
// MAGMA (i.e. CUDA implementation of LAPACK functions)
THC_API void THCTensor_(gels)(THCState *state, THCTensor *rb_, THCTensor *ra_, THCTensor *b_, THCTensor *a_);
THC_API void THCTensor_(geev)(THCState *state, THCTensor *re_, THCTensor *rv_, THCTensor *a_, bool eigenvectors);
THC_API void THCTensor_(potri)(THCState *state, THCTensor *ra_, THCTensor *a, bool upper);
THC_API void THCTensor_(geqrf)(THCState *state, THCTensor *ra_, THCTensor *rtau_, THCTensor *a_);
 
#endif // defined(THC_REAL_IS_FLOAT) || defined(THC_REAL_IS_DOUBLE)
 
#endif