CMakeLists.txt 610 字节
Newer Older
W
Wu Yi 已提交
1
include(operators)
2
register_operators(EXCLUDES fusion_transpose_flatten_concat_op fusion_conv_inception_op fused_fc_elementwise_layernorm_op)
3 4 5
if (WITH_GPU)
  op_library(fusion_transpose_flatten_concat_op)
  file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fusion_transpose_flatten_concat);\n")
6 7 8 9
  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()
10 11
  op_library(fused_fc_elementwise_layernorm_op)
  file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fused_fc_elementwise_layernorm);\n")
12
endif()