forked from ROCm/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists.txt
127 lines (115 loc) · 3.39 KB
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
set(extra_src)
# IF AVX FOUND
IF(C_AVX_FOUND)
LIST(APPEND extra_src ${CMAKE_CURRENT_SOURCE_DIR}/vector/AVX.cpp)
ENDIF(C_AVX_FOUND)
IF(C_AVX2_FOUND)
LIST(APPEND extra_src ${CMAKE_CURRENT_SOURCE_DIR}/vector/AVX2.cpp)
ENDIF(C_AVX2_FOUND)
SET(hdr
THGeneral.h THHalf.h THAllocator.h THSize.h THStorage.h THStorageFunctions.h THTensor.h THTensorApply.h THBlas.h THMath.h
THLapack.h THLogAdd.h THRandom.h THVector.h )
set(ATen_TH_SRCS
${CMAKE_CURRENT_SOURCE_DIR}/THGeneral.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THAllocator.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THSize.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THStorageFunctions.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensor.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorCopy.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorRandom.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorMath.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorMoreMath.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorEvenMoreMath.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorConv.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THTensorLapack.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THBlas.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THLapack.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THLogAdd.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THRandom.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THFile.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THDiskFile.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THMemoryFile.cpp
${CMAKE_CURRENT_SOURCE_DIR}/THVector.cpp
${extra_src}
)
# Remember that PARENT_SCOPE variables are not in the current scope
set(ATen_TH_SRCS ${ATen_TH_SRCS} PARENT_SCOPE)
set(ATen_CPU_SRCS ${ATen_CPU_SRCS} ${ATen_TH_SRCS} PARENT_SCOPE)
######################################################
set(ATen_CPU_INCLUDE ${ATen_CPU_INCLUDE}
${CMAKE_CURRENT_BINARY_DIR}
${CMAKE_CURRENT_SOURCE_DIR}
PARENT_SCOPE)
set(ATen_CUDA_INCLUDE ${ATen_CUDA_INCLUDE}
${CMAKE_CURRENT_BINARY_DIR}
${CMAKE_CURRENT_SOURCE_DIR}
PARENT_SCOPE)
CONFIGURE_FILE(THGeneral.h.in "${CMAKE_CURRENT_BINARY_DIR}/THGeneral.h")
INSTALL(FILES
TH.h
THAllocator.h
THMath.h
THBlas.h
THDiskFile.h
THFile.h
THFilePrivate.h
${CMAKE_CURRENT_BINARY_DIR}/THGeneral.h
THGenerateAllTypes.h
THGenerateDoubleType.h
THGenerateFloatType.h
THGenerateHalfType.h
THGenerateLongType.h
THGenerateIntType.h
THGenerateShortType.h
THGenerateCharType.h
THGenerateByteType.h
THGenerateFloatTypes.h
THGenerateIntTypes.h
THLapack.h
THLogAdd.h
THMemoryFile.h
THRandom.h
THSize.h
THStorage.h
THStorageFunctions.h
THTensor.h
THTensorApply.h
THTensorDimApply.h
THVector.h
THHalf.h
THTensor.hpp
THStorageFunctions.hpp
THGenerator.hpp
DESTINATION "${ATEN_INSTALL_INCLUDE_SUBDIR}/TH")
INSTALL(FILES
vector/AVX.h
vector/AVX2.h
../ATen/native/cpu/avx_mathfun.h
DESTINATION "${ATEN_INSTALL_INCLUDE_SUBDIR}/TH/vector")
INSTALL(FILES
generic/THBlas.cpp
generic/THBlas.h
generic/THLapack.cpp
generic/THLapack.h
generic/THStorage.cpp
generic/THStorage.h
generic/THStorageCopy.cpp
generic/THStorageCopy.h
generic/THTensor.cpp
generic/THTensor.h
generic/THTensor.hpp
generic/THTensorConv.cpp
generic/THTensorConv.h
generic/THTensorCopy.cpp
generic/THTensorCopy.h
generic/THTensorLapack.cpp
generic/THTensorLapack.h
generic/THTensorMath.cpp
generic/THTensorMath.h
generic/THTensorRandom.cpp
generic/THTensorRandom.h
generic/THVectorDispatch.cpp
generic/THVector.h
# See Note [TH abstraction violation]
generic/THTensorFastGetSet.hpp
DESTINATION "${ATEN_INSTALL_INCLUDE_SUBDIR}/TH/generic")