FM / THC

/usr/local/lib64/python3.6/site-packages/torch/include/THC/

/usr/local/lib64/python3.6/site-packages/torch/include/THC

Name	Size	Mode	Actions
generic/	-	0755	rm
THC.h	328	0644	edit dl rm
THCAllocator.h	370	0644	edit dl rm
THCAsmUtils.cuh	3444	0644	edit dl rm
THCAtomics.cuh	13094	0644	edit dl rm
THCCachingHostAllocator.h	1254	0644	edit dl rm
THCDeviceTensor-inl.cuh	11515	0644	edit dl rm
THCDeviceTensor.cuh	16160	0644	edit dl rm
THCDeviceTensorUtils-inl.cuh	4497	0644	edit dl rm
THCDeviceTensorUtils.cuh	2720	0644	edit dl rm
THCDeviceUtils.cuh	942	0644	edit dl rm
THCGeneral.h	2741	0644	edit dl rm
THCGeneral.hpp	743	0644	edit dl rm
THCGenerateAllTypes.h	958	0644	edit dl rm
THCGenerateBFloat16Type.h	515	0644	edit dl rm
THCGenerateBoolType.h	449	0644	edit dl rm
THCGenerateByteType.h	419	0644	edit dl rm
THCGenerateCharType.h	418	0644	edit dl rm
THCGenerateComplexDoubleType.h	533	0644	edit dl rm
THCGenerateComplexFloatType.h	526	0644	edit dl rm
THCGenerateComplexTypes.h	298	0644	edit dl rm
THCGenerateDoubleType.h	464	0644	edit dl rm
THCGenerateFloatType.h	550	0644	edit dl rm
THCGenerateFloatTypes.h	779	0644	edit dl rm
THCGenerateHalfType.h	481	0644	edit dl rm
THCGenerateIntType.h	414	0644	edit dl rm
THCGenerateLongType.h	419	0644	edit dl rm
THCGenerateShortType.h	424	0644	edit dl rm
THCIntegerDivider.cuh	4095	0644	edit dl rm
THCNumerics.cuh	19813	0644	edit dl rm
THCScanUtils.cuh	4789	0644	edit dl rm
THCSleep.h	232	0644	edit dl rm
THCStorage.h	487	0644	edit dl rm
THCStorage.hpp	846	0644	edit dl rm
THCStorageCopy.h	466	0644	edit dl rm
THCTensor.h	623	0644	edit dl rm
THCTensor.hpp	1070	0644	edit dl rm
THCTensorCopy.h	467	0644	edit dl rm
THCTensorCopy.hpp	600	0644	edit dl rm
THCTensorMathReduce.cuh	664	0644	edit dl rm
THCThrustAllocator.cuh	618	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/THC/THCDeviceTensorUtils-inl.cuh (4497B)

namespace detail {

// Add a layer of SFINAE to support static_assert
template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim, bool B>
struct UpcastTHCRoot {
  static THCDeviceTensor<T, NewDim, IndexT, PtrTraits>
  make(THCState* state, THCudaTensor* t);
};

template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim, bool B>
struct UpcastTHC :
      UpcastTHCRoot<T, Dim, IndexT, PtrTraits, NewDim, B> {
};

// Never instantiated SFINAE purposes only
template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim>
struct UpcastTHC<T, Dim, IndexT, PtrTraits, NewDim, false> :
      UpcastTHCRoot<T, Dim, IndexT, PtrTraits, NewDim, false> {
};

template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim>
struct UpcastTHC<T, Dim, IndexT, PtrTraits, NewDim, true> :
      UpcastTHCRoot<T, Dim, IndexT, PtrTraits, NewDim, true>  {
  static THCDeviceTensor<T, NewDim, IndexT, PtrTraits>
  make(THCState* state, THCudaTensor* t) {
    thc_static_assert(NewDim > Dim);
    return toDeviceTensor<T, Dim, IndexT, PtrTraits>(state, t).
      template upcastOuter<NewDim>();
  }
};

// Add a layer of SFINAE to support static_assert
template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim, bool B>
struct DowncastTHCRoot {
  static THCDeviceTensor<T, NewDim, IndexT, PtrTraits>
  make(THCState* state, THCudaTensor* t);
};

template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim, bool B>
struct DowncastTHC :
      DowncastTHCRoot<T, Dim, IndexT, PtrTraits, NewDim, B> {
};

// Never instantiated SFINAE purposes only
template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim>
struct DowncastTHC<T, Dim, IndexT, PtrTraits, NewDim, false> :
      DowncastTHCRoot<T, Dim, IndexT, PtrTraits, NewDim, false> {
};

template <typename T, int Dim, typename IndexT,
          template <typename U> class PtrTraits,
          int NewDim>
struct DowncastTHC<T, Dim, IndexT, PtrTraits, NewDim, true> :
      DowncastTHCRoot<T, Dim, IndexT, PtrTraits, NewDim, true>  {
  static THCDeviceTensor<T, NewDim, IndexT, PtrTraits>
  make(THCState* state, THCudaTensor* t) {
    thc_static_assert(NewDim < Dim);
    return toDeviceTensor<T, Dim, IndexT, PtrTraits>(state, t).
      template downcastOuter<NewDim>();
  }
};

} // namespace detail

#define SWITCH_UNROLL_CUDA_CAST_FACTORY(i)                              \
  case i:                                                               \
  if (NewDim > i) {                                                     \
    return detail::UpcastTHC<T, i, IndexT,                              \
                             PtrTraits, NewDim, (NewDim > i)>::         \
      make(state, t);                                                   \
  } else if (NewDim == i) {                                             \
    return toDeviceTensor<T, NewDim, IndexT, PtrTraits>(state, t);      \
  } else {                                                              \
    return detail::DowncastTHC<T, i, IndexT,                            \
                               PtrTraits, NewDim, (NewDim < i)>::       \
      make(state, t);                                                   \
  }                                                                     \
  /* break; */

template <typename T, int NewDim,
          typename IndexT, template <typename U> class PtrTraits>
THCDeviceTensor<T, NewDim, IndexT, PtrTraits>
toDeviceTensorCast(THCState* state, THCudaTensor* t) {
  switch (THCudaTensor_nDimensionLegacyAll(state, t)) {
    SWITCH_UNROLL_CUDA_CAST_FACTORY(1);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(2);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(3);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(4);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(5);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(6);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(7);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(8);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(9);
    SWITCH_UNROLL_CUDA_CAST_FACTORY(10);
    default:
      ;
  }

// Not implemented
  THError("THCDeviceTensor dimension size not supported");
  return NULL; /* never enters this piece, appeasing compiler warnings */
}

#undef SWITCH_UNROLL_CUDA_CAST_FACTORY

namespace detail { // Add a layer of SFINAE to support static_assert template class PtrTraits, int NewDim, bool B> struct UpcastTHCRoot { static THCDeviceTensor make(THCState* state, THCudaTensor* t); }; template class PtrTraits, int NewDim, bool B> struct UpcastTHC : UpcastTHCRoot { }; // Never instantiated SFINAE purposes only template class PtrTraits, int NewDim> struct UpcastTHC : UpcastTHCRoot { }; template class PtrTraits, int NewDim> struct UpcastTHC : UpcastTHCRoot { static THCDeviceTensor make(THCState* state, THCudaTensor* t) { thc_static_assert(NewDim > Dim); return toDeviceTensor(state, t). template upcastOuter(); } }; // Add a layer of SFINAE to support static_assert template class PtrTraits, int NewDim, bool B> struct DowncastTHCRoot { static THCDeviceTensor make(THCState* state, THCudaTensor* t); }; template class PtrTraits, int NewDim, bool B> struct DowncastTHC : DowncastTHCRoot { }; // Never instantiated SFINAE purposes only template class PtrTraits, int NewDim> struct DowncastTHC : DowncastTHCRoot { }; template class PtrTraits, int NewDim> struct DowncastTHC : DowncastTHCRoot { static THCDeviceTensor make(THCState* state, THCudaTensor* t) { thc_static_assert(NewDim < Dim); return toDeviceTensor(state, t). template downcastOuter(); } }; } // namespace detail #define SWITCH_UNROLL_CUDA_CAST_FACTORY(i) \ case i: \ if (NewDim > i) { \ return detail::UpcastTHC i)>:: \ make(state, t); \ } else if (NewDim == i) { \ return toDeviceTensor(state, t); \ } else { \ return detail::DowncastTHC:: \ make(state, t); \ } \ /* break; */ template class PtrTraits> THCDeviceTensor toDeviceTensorCast(THCState* state, THCudaTensor* t) { switch (THCudaTensor_nDimensionLegacyAll(state, t)) { SWITCH_UNROLL_CUDA_CAST_FACTORY(1); SWITCH_UNROLL_CUDA_CAST_FACTORY(2); SWITCH_UNROLL_CUDA_CAST_FACTORY(3); SWITCH_UNROLL_CUDA_CAST_FACTORY(4); SWITCH_UNROLL_CUDA_CAST_FACTORY(5); SWITCH_UNROLL_CUDA_CAST_FACTORY(6); SWITCH_UNROLL_CUDA_CAST_FACTORY(7); SWITCH_UNROLL_CUDA_CAST_FACTORY(8); SWITCH_UNROLL_CUDA_CAST_FACTORY(9); SWITCH_UNROLL_CUDA_CAST_FACTORY(10); default: ; } // Not implemented THError("THCDeviceTensor dimension size not supported"); return NULL; /* never enters this piece, appeasing compiler warnings */ } #undef SWITCH_UNROLL_CUDA_CAST_FACTORY