CMakeLists.txt

core_gather_headers()

gather_srcs(cinnapi_src SRCS
    remove_nested_block.cc
    replace_call_with_expr.cc
    ir_copy.cc
    ir_replace.cc
    replace_var_with_expr.cc
    tensor_write_tell.cc
    ir_simplify.cc
    optimize.cc
    vectorize_loops.cc
    unroll_loops.cc
    transform_polyfor_to_for.cc
    eliminate_broadcast_in_forloop.cc
    fold_cinn_call_arguments.cc
    call_arg_list_to_pod_value.cc
    insert_debug_log_callee.cc
    lower_function_call_bind_vars.cc
    extern_call_process.cc
    map_extern_call.cc
    compute_inline_expand.cc
    buffer_assign.cc
    replace_const_param_to_integer.cc
    cast_simplify.cc
    if_simplify.cc
    lower_intrin.cc
    cast_bool_to_int8.cc
    collect_undefined_vars.cc
    var_mod_simplify.cc
    remove_schedule_block.cc
    )

if (WITH_CUDA)
  gather_srcs(cinnapi_src SRCS transform_gpu_forloop.cc)
endif()


cinn_cc_test(test_remove_nested_block SRCS remove_nested_block_test.cc DEPS cinncore)
cinn_cc_test(test_ir_copy SRCS ir_copy_test.cc DEPS cinncore)
cinn_cc_test(test_ir_simplify SRCS ir_simplify_test.cc DEPS cinncore)
cinn_cc_test(test_replace_call_with_expr SRCS replace_call_with_expr_test.cc DEPS cinncore)
cinn_cc_test(test_vectorize_loops SRCS vectorize_loops_test.cc DEPS cinncore ARGS ${global_test_args})
cinn_cc_test(test_transform_polyfor_to_for SRCS transform_polyfor_to_for_test.cc DEPS cinncore ARGS ${global_test_args})
cinn_cc_test(test_optimize SRCS optimize_test.cc DEPS cinncore)
cinn_cc_test(test_cache_read_write_replace SRCS cache_read_write_replace_test.cc DEPS cinncore)
cinn_cc_test(test_cast_simplify SRCS cast_simplify_test.cc DEPS cinncore)
cinn_cc_test(test_if_simplify SRCS if_simplify_test.cc DEPS cinncore)
cinn_cc_test(test_remove_schedule_block SRCS remove_schedule_block_test.cc DEPS cinncore)
cinn_cc_test(test_unroll_loops SRCS unroll_loops_test.cc DEPS cinncore)