#ifndef THC_GENERIC_FILE #define THC_GENERIC_FILE "generic/common.cuh" #else TensorInfo THCTensor_(getTensorInfo)(THCState *state, THCTensor *tensor) { TensorInfo tensorInfo = TensorInfo(); tensorInfo.data = THCTensor_(data)(state, tensor); tensorInfo.dims = THCTensor_(nDimension)(state, tensor); for (ptrdiff_t d = 0; d < tensorInfo.dims; d++) { tensorInfo.size[d] = THCTensor_(size)(state, tensor, d); tensorInfo.stride[d] = THCTensor_(stride)(state, tensor, d); } return tensorInfo; } #endif // THC_GENERIC_FILE