You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
2.2 KiB
57 lines
2.2 KiB
include(operators)
|
|
register_operators(EXCLUDES
|
|
fused_bn_activation_op
|
|
conv_fusion_op
|
|
fusion_transpose_flatten_concat_op
|
|
fusion_conv_inception_op
|
|
fused_fc_elementwise_layernorm_op
|
|
multihead_matmul_op
|
|
fused_embedding_eltwise_layernorm_op
|
|
fusion_group_op
|
|
fusion_gru_op
|
|
fused_bn_add_activation_op)
|
|
|
|
# fusion_gru_op does not have CUDA kernel
|
|
op_library(fusion_gru_op)
|
|
file(APPEND ${pybind_file} "USE_CPU_ONLY_OP(fusion_gru);\n")
|
|
|
|
if (WITH_GPU)
|
|
# fused_bn_activation_op needs cudnn 7.4.1 above
|
|
if (NOT ${CUDNN_VERSION} VERSION_LESS 7401)
|
|
op_library(fused_bn_activation_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fused_batch_norm_act);\n")
|
|
endif()
|
|
# conv_fusion_op needs cudnn 7 above
|
|
if (NOT ${CUDNN_VERSION} VERSION_LESS 7100)
|
|
op_library(conv_fusion_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(conv2d_fusion);\n")
|
|
endif()
|
|
# fusion_transpose_flatten_concat_op
|
|
op_library(fusion_transpose_flatten_concat_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fusion_transpose_flatten_concat);\n")
|
|
# fusion_conv_inception_op needs cudnn 7 above
|
|
if (NOT ${CUDNN_VERSION} VERSION_LESS 7100)
|
|
op_library(fusion_conv_inception_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(conv2d_inception_fusion);\n")
|
|
endif()
|
|
# fused_fc_elementwise_layernorm_op
|
|
op_library(fused_fc_elementwise_layernorm_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fused_fc_elementwise_layernorm);\n")
|
|
# multihead_matmul_op
|
|
op_library(multihead_matmul_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(multihead_matmul);\n")
|
|
op_library(fused_embedding_eltwise_layernorm_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fused_embedding_eltwise_layernorm);\n")
|
|
# fusion_group
|
|
if(NOT APPLE AND NOT WIN32)
|
|
op_library(fusion_group_op DEPS device_code)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fusion_group);\n")
|
|
cc_test(test_fusion_group_op SRCS fusion_group_op_test.cc DEPS fusion_group_op)
|
|
endif()
|
|
# fused_bn_add_activation
|
|
if (NOT ${CUDNN_VERSION} VERSION_LESS 7401)
|
|
op_library(fused_bn_add_activation_op)
|
|
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fused_bn_add_activation);\n")
|
|
endif()
|
|
endif()
|