# Fusion Group IR Pass Tests

cc_test(
  test_fusion_group_pass
  SRCS fusion_group_pass_test.cc
  DEPS fusion_group_pass graph_viz_pass)

if(WITH_GPU OR WITH_ROCM)
  cc_test(
    test_code_generator
    SRCS code_generator_test.cc
    DEPS code_generator phi common lod_tensor graph_viz_pass)

  # Set timeout for test_code_generator
  if(WITH_TESTING AND TEST test_code_generator)
    set_tests_properties(test_code_generator PROPERTIES TIMEOUT 120)
  endif()
endif()
