# Copyright (c) 2021 Graphcore Ltd. All rights reserved.
# Tests of pipelining
# to run only these tests with ctest, consider using something like,
# `ctest -R pipeline -j5 -VV'.

add_unit_test(pipeline_topo_con_test_0 pipeline_topo_con_test_0.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_continuous_equivalent_test_0
                         pipeline_continuous_equivalent_test_0.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_discontiguous_ipu_copy_test_0
                         pipeline_discontiguous_ipu_copy_test_0.cpp  VARIANTS "IpuModel2")
add_unit_test(pipeline_discontiguous_ipu_copy_test_1
                         pipeline_discontiguous_ipu_copy_test_1.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_quadratic_epsilon_test_0
                         pipeline_quadratic_epsilon_test_0.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_quadratic_epsilon_test_1
                         pipeline_quadratic_epsilon_test_1.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_recompute_numerical_test_0
                         pipeline_recompute_numerical_test_0.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_anchor_recomputed_tensor_check
                         pipeline_anchor_recomputed_tensor_check.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_no_multi_source_copies_test_0
                         pipeline_no_multi_source_copies_test_0.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_recompute_ir_test_0
                         pipeline_recompute_ir_test_0.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_recompute_ir_test_1
                         pipeline_recompute_ir_test_1.cpp VARIANTS "IpuModel2")
add_unit_test(pipeline_recompute_ir_test_2
                         pipeline_recompute_ir_test_2.cpp VARIANTS "IpuModel2")

add_popart_py_unit_test(pipeline_full_recompute_test VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_grad_accl_test VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_multi_loss_numerical_0 VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_test MATCHEXPR "test_output_matches_train" VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_test MATCHEXPR "test_output_matches_infer" VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_test MATCHEXPR "not test_output_matches_train and not test_output_matches_infer" VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_dropout_test VARIANTS Hw)
add_popart_py_unit_test(replicated_pipeline_test VARIANTS "${HW_EXCLUDE_C600}")
add_popart_py_unit_test(pipeline_boundary_test VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_consecutive_stages_test VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_multiple_restores_per_stash VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_grad_tensor_consumed_by_multiple_pipeline_stages VARIANTS IpuModel2)
add_popart_py_unit_test(pipeline_user_recompute_test VARIANTS IpuModel2)
add_popart_py_unit_test(explicit_pipeline_test VARIANTS IpuModel2)