File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -97,7 +97,7 @@ sequence_pooling segment_pooling executor device_memory_aligment generator)
97
97
set (COMMON_OP_DEPS ${COMMON_OP_DEPS} dynload_warpctc )
98
98
set (COMMON_OP_DEPS ${COMMON_OP_DEPS} sequence_padding sequence_scale cos_sim_functor memory jit_kernel_helper concat_and_split cross_entropy softmax vol2col im2col sampler sample_prob tree2col )
99
99
set (COMMON_OP_DEPS ${COMMON_OP_DEPS} sequence2batch lstm_compute matrix_bit_code gru_compute activation_functions beam_search fc matrix_inverse )
100
- set (COMMON_OP_DEPS ${COMMON_OP_DEPS} box_wrapper boost )
100
+ set (COMMON_OP_DEPS ${COMMON_OP_DEPS} box_wrapper boost ps_gpu_wrapper )
101
101
set (COMMON_OP_DEPS ${COMMON_OP_DEPS} common_infer_shape_functions )
102
102
if (WITH_GPU )
103
103
set (COMMON_OP_DEPS ${COMMON_OP_DEPS} depthwise_conv prelu bert_encoder_functor )
You can’t perform that action at this time.
0 commit comments