File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -34,7 +34,12 @@ if (WITH_GPU AND TENSORRT_FOUND)
34
34
add_subdirectory (tensorrt )
35
35
endif ()
36
36
37
- register_operators (EXCLUDES warpctc_op conv_fusion_op )
37
+ SET (OP_HEADER_DEPS xxhash )
38
+ if (WITH_GPU )
39
+ SET (OP_HEADER_DEPS ${OP_HEADER_DEPS} cub )
40
+ endif ()
41
+
42
+ register_operators (EXCLUDES warpctc_op conv_fusion_op DEPS ${OP_HEADER_DEPS} )
38
43
39
44
# warpctc_cudnn need cudnn 7 above
40
45
if (WITH_GPU )
@@ -49,14 +54,14 @@ else()
49
54
op_library (warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale )
50
55
endif ()
51
56
52
- set (COMMON_OP_DEPS "" )
57
+ set (COMMON_OP_DEPS ${OP_HEADER_DEPS} )
53
58
54
- set (COMMON_OP_DEPS ${COMMON_OP_DEPS} xxhash selected_rows_functor selected_rows lod_tensor maxouting unpooling pooling lod_rank_table context_project sequence_pooling executor dynload_warpctc sequence_padding sequence_scale cos_sim_functor memory jit_kernel concat_and_split cross_entropy softmax vol2col im2col sampler )
59
+ set (COMMON_OP_DEPS ${COMMON_OP_DEPS} selected_rows_functor selected_rows lod_tensor maxouting unpooling pooling lod_rank_table context_project sequence_pooling executor dynload_warpctc sequence_padding sequence_scale cos_sim_functor memory jit_kernel concat_and_split cross_entropy softmax vol2col im2col sampler )
55
60
if (NOT WIN32 )
56
61
set (COMMON_OP_DEPS ${COMMON_OP_DEPS} sequence2batch lstm_compute matrix_bit_code gru_compute activation_functions )
57
62
endif ()
58
63
if (WITH_GPU )
59
- set (COMMON_OP_DEPS ${COMMON_OP_DEPS} depthwise_conv cub )
64
+ set (COMMON_OP_DEPS ${COMMON_OP_DEPS} depthwise_conv )
60
65
endif ()
61
66
62
67
# FIXME(typhoonzero): operator deps may not needed.
You can’t perform that action at this time.
0 commit comments