forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists.txt
84 lines (82 loc) · 3.92 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
SET(ATen_CUDA_SRCS ${ATen_CUDA_SRCS}
${CMAKE_CURRENT_SOURCE_DIR}/AbsCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/Abs.cu
${CMAKE_CURRENT_SOURCE_DIR}/BatchNormalization.cu
${CMAKE_CURRENT_SOURCE_DIR}/BCECriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/ClassNLLCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/Col2Im.cu
${CMAKE_CURRENT_SOURCE_DIR}/DistKLDivCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/ELU.cu
${CMAKE_CURRENT_SOURCE_DIR}/FeatureLPPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/GatedLinearUnit.cu
${CMAKE_CURRENT_SOURCE_DIR}/HardTanh.cu
${CMAKE_CURRENT_SOURCE_DIR}/Im2Col.cu
${CMAKE_CURRENT_SOURCE_DIR}/IndexLinear.cu
${CMAKE_CURRENT_SOURCE_DIR}/L1Cost.cu
${CMAKE_CURRENT_SOURCE_DIR}/LeakyReLU.cu
${CMAKE_CURRENT_SOURCE_DIR}/LogSigmoid.cu
${CMAKE_CURRENT_SOURCE_DIR}/LookupTableBag.cu
${CMAKE_CURRENT_SOURCE_DIR}/LookupTable.cu
${CMAKE_CURRENT_SOURCE_DIR}/MarginCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/MSECriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/MultiLabelMarginCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/MultiMarginCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/RReLU.cu
${CMAKE_CURRENT_SOURCE_DIR}/Sigmoid.cu
${CMAKE_CURRENT_SOURCE_DIR}/SmoothL1Criterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/SoftMarginCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/SoftPlus.cu
${CMAKE_CURRENT_SOURCE_DIR}/SoftShrink.cu
${CMAKE_CURRENT_SOURCE_DIR}/SparseLinear.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialAdaptiveAveragePooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialAdaptiveMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialAveragePooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialClassNLLCriterion.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialConvolutionLocal.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialConvolutionMM.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialCrossMapLRN.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialDepthwiseConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialDilatedConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialDilatedMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialFractionalMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialFullConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialFullDilatedConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialMaxUnpooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialReflectionPadding.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialReplicationPadding.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialSubSampling.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialUpSamplingBilinear.cu
${CMAKE_CURRENT_SOURCE_DIR}/SpatialUpSamplingNearest.cu
${CMAKE_CURRENT_SOURCE_DIR}/Sqrt.cu
${CMAKE_CURRENT_SOURCE_DIR}/Square.cu
${CMAKE_CURRENT_SOURCE_DIR}/Tanh.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalReflectionPadding.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalReplicationPadding.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalRowConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalUpSamplingLinear.cu
${CMAKE_CURRENT_SOURCE_DIR}/TemporalUpSamplingNearest.cu
${CMAKE_CURRENT_SOURCE_DIR}/Threshold.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricAdaptiveAveragePooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricAdaptiveMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricAveragePooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricDilatedConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricDilatedMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricFractionalMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricFullConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricFullDilatedConvolution.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricMaxPooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricMaxUnpooling.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricReplicationPadding.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricUpSamplingNearest.cu
${CMAKE_CURRENT_SOURCE_DIR}/VolumetricUpSamplingTrilinear.cu
PARENT_SCOPE)
set(ATen_CUDA_INCLUDE ${ATen_CUDA_INCLUDE}
"${CMAKE_CURRENT_SOURCE_DIR}"
PARENT_SCOPE)
install(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
DESTINATION ${ATEN_INSTALL_INCLUDE_SUBDIR}
FILES_MATCHING PATTERN "*.h" PATTERN "*.cuh")