|
|
|
@ -37,7 +37,13 @@ if (WITH_GPU)
|
|
|
|
|
SET(OP_HEADER_DEPS ${OP_HEADER_DEPS} cub)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
register_operators(EXCLUDES warpctc_op conv_fusion_op lookup_table_op DEPS ${OP_HEADER_DEPS})
|
|
|
|
|
SET(OP_PREFETCH_DEPS "")
|
|
|
|
|
if (WITH_DISTRIBUTE)
|
|
|
|
|
SET(OP_PREFETCH_DEPS ${OP_PREFETCH_DEPS} parameter_prefetch)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
register_operators(EXCLUDES warpctc_op conv_fusion_op DEPS ${OP_HEADER_DEPS} ${OP_PREFETCH_DEPS})
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# warpctc_op needs cudnn 7 above
|
|
|
|
|
if (WITH_GPU AND NOT WIN32)
|
|
|
|
@ -55,8 +61,6 @@ else()
|
|
|
|
|
op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
op_library(lookup_table_op DEPS parameter_prefetch)
|
|
|
|
|
|
|
|
|
|
set(COMMON_OP_DEPS ${OP_HEADER_DEPS})
|
|
|
|
|
|
|
|
|
|
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)
|
|
|
|
|