if(WITH_TESTING AND WITH_CINN)
  paddle_test(test_pir_compiler SRCS pir_compiler_test.cc DEPS pir_compiler
              cinn_runtime_dialect)
  set_tests_properties(test_pir_compiler PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(test_jit_instruction SRCS jit_instruction_test.cc DEPS
              cinn_runtime_dialect pir_compiler)
  set_tests_properties(test_jit_instruction PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(
    dialect_convert_test
    SRCS
    dialect_convert_test.cc
    DEPS
    drr
    pd_to_cinn_pass
    op_dialect_vjp
    cinn_op_dialect
    pir)
  set_tests_properties(dialect_convert_test PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(
    add_broadcast_to_elementwise_test
    SRCS
    add_broadcast_to_elementwise_test.cc
    DEPS
    drr
    pd_to_cinn_pass
    op_dialect_vjp
    cinn_op_dialect
    add_broadcast_to_elementwise_pass
    pir)
  set_tests_properties(dialect_convert_test PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(
    test_sub_graph_extract
    SRCS
    sub_graph_extract_test.cc
    DEPS
    drr
    pd_to_cinn_pass
    op_dialect_vjp
    cinn_op_dialect
    pir_transforms
    pir)
  set_tests_properties(test_sub_graph_extract PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(
    ir_op_fusion_test
    SRCS
    ir_op_fusion_test.cc
    DEPS
    op_with_group_merge_pass
    op_dialect_vjp
    cinn_op_dialect
    pir)
  set_tests_properties(ir_op_fusion_test PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(
    test_pir_all_path
    SRCS
    pir_all_path_test.cc
    DEPS
    op_with_group_merge_pass
    pir_transforms
    cinn_op_dialect
    pd_to_cinn_pass
    add_broadcast_to_elementwise_pass)
  set_tests_properties(test_pir_all_path PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(
    test_group_op
    SRCS
    group_op_test.cc
    DEPS
    pd_to_cinn_pass
    add_broadcast_to_elementwise_pass
    op_with_group_merge_pass
    cinn_op_dialect
    pir_transforms)
  set_tests_properties(test_group_op PROPERTIES LABELS "RUN_TYPE=CINN")

  paddle_test(test_pir_build_cinn_pass SRCS build_cinn_pass_test.cc DEPS
              pir_transforms pir)
  set_tests_properties(test_pir_build_cinn_pass PROPERTIES LABELS
                                                           "RUN_TYPE=CINN")

  paddle_test(test_compilation_task SRCS compilation_task_test.cc DEPS pir)
  set_tests_properties(test_compilation_task PROPERTIES LABELS "RUN_TYPE=CINN")
endif()
