diff options
-rw-r--r-- | lib/TH/CMakeLists.txt | 5 | ||||
-rw-r--r-- | lib/TH/TH.h | 1 | ||||
-rw-r--r-- | lib/TH/THSize.c | 11 | ||||
-rw-r--r-- | lib/TH/THSize.h | 8 |
4 files changed, 23 insertions, 2 deletions
diff --git a/lib/TH/CMakeLists.txt b/lib/TH/CMakeLists.txt index 22cdf33..dd0b75d 100644 --- a/lib/TH/CMakeLists.txt +++ b/lib/TH/CMakeLists.txt @@ -250,11 +250,11 @@ IF(C_AVX2_FOUND) ENDIF(C_AVX2_FOUND) SET(hdr - THGeneral.h THHalf.h THAllocator.h THStorage.h THTensor.h THTensorApply.h THBlas.h THMath.h + THGeneral.h THHalf.h THAllocator.h THSize.h THStorage.h THTensor.h THTensorApply.h THBlas.h THMath.h THLapack.h THLogAdd.h THRandom.h THVector.h THAtomic.h ) SET(src - THGeneral.c THHalf.c THAllocator.c THStorage.c THTensor.c THBlas.c THLapack.c + THGeneral.c THHalf.c THAllocator.c THSize.c THStorage.c THTensor.c THBlas.c THLapack.c THLogAdd.c THRandom.c THFile.c THDiskFile.c THMemoryFile.c THAtomic.c THVector.c) SET(src ${src} ${hdr} ${simd}) @@ -401,6 +401,7 @@ INSTALL(FILES THLogAdd.h THMemoryFile.h THRandom.h + THSize.h THStorage.h THTensor.h THTensorApply.h diff --git a/lib/TH/TH.h b/lib/TH/TH.h index cdf331d..11f208c 100644 --- a/lib/TH/TH.h +++ b/lib/TH/TH.h @@ -12,6 +12,7 @@ #include "THVector.h" #include "THLogAdd.h" #include "THRandom.h" +#include "THSize.h" #include "THStorage.h" #include "THTensor.h" #include "THTensorApply.h" diff --git a/lib/TH/THSize.c b/lib/TH/THSize.c new file mode 100644 index 0000000..f95d333 --- /dev/null +++ b/lib/TH/THSize.c @@ -0,0 +1,11 @@ +int THLongStorage_isSameSizeAs(const long *sizeA, long dimsA, const long *sizeB, long dimsB) { + int d; + if (dimsA != dimsB) + return 0; + for(d = 0; d < dimsA; ++d) + { + if(sizeA[d] != sizeB[d]) + return 0; + } + return 1; +} diff --git a/lib/TH/THSize.h b/lib/TH/THSize.h new file mode 100644 index 0000000..e582977 --- /dev/null +++ b/lib/TH/THSize.h @@ -0,0 +1,8 @@ +#ifndef TH_SIZE_INC +#define TH_SIZE_INC + +#include "THGeneral.h" + +TH_API int THSize_isSameSizeAs(const long *sizeA, long dimsA, const long *sizeB, long dimsB); + +#endif |