|
|
|
@ -9,6 +9,6 @@ cc_library(fusion_group_pass
|
|
|
|
SRCS fusion_group_pass.cc elementwise_group_detector.cc
|
|
|
|
SRCS fusion_group_pass.cc elementwise_group_detector.cc
|
|
|
|
DEPS subgraph_detector fuse_pass_base code_generator device_code)
|
|
|
|
DEPS subgraph_detector fuse_pass_base code_generator device_code)
|
|
|
|
cc_test(test_fusion_group_pass SRCS fusion_group_pass_tester.cc DEPS fusion_group_pass graph_viz_pass)
|
|
|
|
cc_test(test_fusion_group_pass SRCS fusion_group_pass_tester.cc DEPS fusion_group_pass graph_viz_pass)
|
|
|
|
if(WITH_GPU AND NOT ON_INFER)
|
|
|
|
if(WITH_TESTING)
|
|
|
|
set_tests_properties(test_code_generator PROPERTIES TIMEOUT 120)
|
|
|
|
set_tests_properties(test_code_generator PROPERTIES TIMEOUT 120)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|