CMakeLists.txt 666 字节
Newer Older
1 2 3
cc_library(code_generator
    SRCS operation.cc code_generator.cc code_generator_helper.cc
    DEPS graph subgraph_detector)
4
if(WITH_GPU OR WITH_ROCM)
5
    cc_test(test_code_generator SRCS code_generator_tester.cc DEPS code_generator device_code lod_tensor graph_viz_pass)
6 7 8 9
endif()

cc_library(fusion_group_pass
    SRCS fusion_group_pass.cc elementwise_group_detector.cc
10
    DEPS subgraph_detector fuse_pass_base code_generator device_code)
11
cc_test(test_fusion_group_pass SRCS fusion_group_pass_tester.cc DEPS fusion_group_pass graph_viz_pass)
12
if(WITH_TESTING AND TEST test_code_generator)
13
    set_tests_properties(test_code_generator PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
14
endif()