# Copyright (c) 2021 Graphcore Ltd. All rights reserved.
add_unit_test(accumulateouterfragmentparallelizertest accumulateouterfragmentparallelizer_test.cpp VARIANTS "IpuModel2")
add_unit_test(aliaszerocopytest aliaszerocopy_test.cpp)
add_unit_test(aliaszerocopytest1 aliaszerocopy_test1.cpp)
add_unit_test(allocatortest allocator_test.cpp)
add_unit_test(boollogictest boollogic_test.cpp)
add_unit_test(builderpartialstest builder_partials_test.cpp)
add_unit_test(casttest cast_test.cpp)
add_unit_test(collectivestest collectives_test.cpp VARIANTS "${HW_EXCLUDE_C600}")
add_unit_test(copy_named_buffers_to_device_test copy_named_buffers_to_device_test.cpp)
add_unit_test(custompatterntest custom_pattern_test.cpp)
add_unit_test(customtransformtest custom_transform_test.cpp)
add_unit_test(dataflowtest dataflowtest.cpp)
add_unit_test(decomposegradientsummationtest decompose_gradient_summation_test.cpp)
add_unit_test(executable_serialization_test executable_serialization_test.cpp VARIANTS "${HW_EXCLUDE_C600}")
add_unit_test(graphedgemaptest graph_edgemap_test.cpp SUPPORT_LIBS test-graphs-test-util)
add_unit_test(exceptiontest exceptiontest.cpp)
add_unit_test(inputshapeinfotest inputshapeinfotest.cpp)
add_unit_test(iosubgraphcostmodel iosubgraphcostmodel_test.cpp)
add_unit_test(irhashtest ir_hash_test.cpp VARIANTS "IpuModel2")
add_unit_test(isnonlinearitytest is_nonlinearity_test.cpp)
add_unit_test(isnormtest is_norm_test.cpp)
add_unit_test(loggingtest loggingtest.cpp)
add_unit_test(maxcliquetest maxclique_test.cpp)
add_unit_test(mergecopiestest mergecopies_test.cpp)
add_unit_test(nogradoptest no_gradop_test.cpp)
add_unit_test(numpybroadcastshapetest numpybroadcastshapetest.cpp)
add_unit_test(opmanagertest op_manager_test.cpp)
add_unit_test(opxtensoraliasingtest opx_tensor_aliasing_test.cpp)
add_unit_test(outliningirtest outlining_ir_test.cpp)
add_unit_test(poprithmstransitiveclosuretest poprithmstransitiveclosure_test.cpp SUPPORT_LIBS test-graphs-test-util)
add_unit_test(prunetest prune_test.cpp SUPPORT_LIBS test-graphs-test-util)
add_unit_test(simple_addition_test simple_addition_test.cpp)
add_unit_test(subgraph_partitioning_test subgraph_partitioning_test.cpp)
add_unit_test(syncpatterntest sync_pattern_test.cpp VARIANTS "Hw" PROPERTIES RUN_SERIAL TRUE)
add_unit_test(syntheticdatatest synthetic_data_test.cpp)
add_unit_test(transformtest transform_test.cpp)
add_unit_test(vertex_vgid_test vertex_vgid_test.cpp VARIANTS "IpuModel2")
add_unit_test(viewchangingtest view_changing_test.cpp)
add_unit_test(debuginfotest debug_info_test.cpp)
add_unit_test(deprecated_session_options_cpp deprecated_session_options.cpp)

add_unit_test(constop_test constop_test.cpp)
add_unit_test(opsetcheck_test opset_check_test.cpp)
add_unit_test(preplan_convolutions_test preplan_convolutions_test.cpp VARIANTS "IpuModel2")
add_unit_test(preplan_matmuls_test preplan_matmuls_test.cpp VARIANTS "IpuModel2")
add_unit_test(rng_mapping_test rng_mapping_test.cpp)

add_unit_test(inittensoroffsetmap_test inittensoroffsetmap_test.cpp VARIANTS Hw)

# add_popart_py_unit_test(test_util) Utility
# add_popart_py_unit_test(test_session) Utility

# Python tests. Please try to sort alphabetically.
add_popart_py_unit_test(abort_test VARIANTS Hw)
add_popart_py_unit_test(addbias)
add_popart_py_unit_test(aliaszerocopy_model_test VARIANTS IpuModel2)
add_popart_py_unit_test(all_constexpr)
add_popart_py_unit_test(anchorreturntype_test VARIANTS IpuModel2)
add_popart_py_unit_test(annotations_test)
add_popart_py_unit_test(auto_loss_scaling_gradient_accumulation_test)
add_popart_py_unit_test(auto_loss_scaling_identical_weight_updates_0 VARIANTS IpuModel2)
add_popart_py_unit_test(auto_loss_scaling_identical_weight_updates_1 VARIANTS IpuModel2)
add_popart_py_unit_test(auto_virtual_graph_test VARIANTS IpuModel2)
add_popart_py_unit_test(builder_name_test)
add_popart_py_unit_test(builder_test)
add_popart_py_unit_test(clip_by_norm_test)
add_popart_py_unit_test(replica_sharded_with_optimizer VARIANTS IpuModel2)
add_popart_py_unit_test(convolution_options_test VARIANTS IpuModel2)
add_popart_py_unit_test(context_scope_test)
add_popart_py_unit_test(contiguous_memory_test VARIANTS IpuModel2)
add_popart_py_unit_test(available_memory_proportion_test VARIANTS IpuModel2)
add_popart_py_unit_test(bert_test VARIANTS IpuModel2)
add_popart_py_unit_test(cycle_count_test VARIANTS Hw PROPERTIES RUN_SERIAL TRUE)
add_popart_py_unit_test(conv_dithering_test VARIANTS IpuModel2)

add_popart_py_unit_test(decompose_gradient_summation_test VARIANTS IpuModel2)
# device_test should be run serially as it tests that it can attach to each
# IPU available on the machine it is run on.
add_popart_py_unit_test(device_test VARIANTS Hw PROPERTIES RUN_SERIAL TRUE)
add_popart_py_unit_test(dim_param_test)
add_popart_py_unit_test(distributed_test VARIANTS IpuModel2)
# TODO T45747 Requires 2xPOD16
add_popart_py_unit_test(distributed_replicated_graph_test VARIANTS Hw PROPERTIES DISABLED TRUE)
add_popart_py_unit_test(doc_test)
add_popart_py_unit_test(dont_inplace_test)
add_popart_py_unit_test(enhanced_debug)
add_popart_py_unit_test(exception_test VARIANTS Hw)
add_popart_py_unit_test(export_test)
add_popart_py_unit_test(float_to_half_conversion_test)
add_popart_py_unit_test(fp16_test)
# The gradient_accumulation_test file is ran in separate ctest tests as they take a long time.
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_gradient_accumulation_base VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_gradient_accumulation_multi_batch VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_gradient_accumulation_multi_ipu VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_gradient_accumulation_error_inference VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_gradient_accumulation_error_accum_factor_invalid VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_gradient_accumulation_model_proto VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_loading_saved_gradient_accumulationt_tensors VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_adam_gradient_accumulation_base VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_adam_gradient_accumulation_multi_batch VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_adam_gradient_accumulation_multi_ipu VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_adam_gradient_accumulation_model_proto VARIANTS IpuModel2)
add_popart_py_unit_test(gradient_accumulation_test MATCHEXPR test_adam_loading_saved_gradient_accumulationt_tensors VARIANTS IpuModel2)
add_popart_py_unit_test(graph_caching_test VARIANTS IpuModel2)
add_popart_py_unit_test(import_test)
add_popart_py_unit_test(ipu_copy_test VARIANTS IpuModel2)
add_popart_py_unit_test(loader_test)
add_popart_py_unit_test(loss_scaling_test VARIANTS IpuModel2)
add_popart_py_unit_test(loss_scaling_update_period0_test NUM_WORKERS auto VARIANTS IpuModel2)
add_popart_py_unit_test(loss_scaling_update_period1_test NUM_WORKERS auto VARIANTS IpuModel2)
add_popart_py_unit_test(loss_scaling_infrequent_oversteps VARIANTS IpuModel2)
add_popart_py_unit_test(net_test)
add_popart_py_unit_test(np_memory_layout_test)
add_popart_py_unit_test(offline_compilation)
add_popart_py_unit_test(optimizer_test)
add_popart_py_unit_test(options_test)
add_popart_py_unit_test(outlining_test)
add_popart_py_unit_test(outlining_execution_context_test VARIANTS "${HW_EXCLUDE_C600}")
add_popart_py_unit_test(padsumpattern)
add_popart_py_unit_test(partials_tests VARIANTS IpuModel2)
add_popart_py_unit_test(patterns_test)
add_popart_py_unit_test(prefetch_test VARIANTS Hw)
add_popart_py_unit_test(printtensor)
add_popart_py_unit_test(prune_all_error_test)
add_popart_py_unit_test(random_test VARIANTS Hw)
add_popart_py_unit_test(randomsetup_test VARIANTS Hw)
add_popart_py_unit_test(recompute_compatibility_test)
add_popart_py_unit_test(report_test VARIANTS IpuModel2)
add_popart_py_unit_test(report_test_cpu)
add_popart_py_unit_test(saved_executable VARIANTS Hw)
add_popart_py_unit_test(saved_executable_cpu)
add_popart_py_unit_test(serialise_matmul_manually)
add_popart_py_unit_test(stream_test)
add_popart_py_unit_test(stubs_test)
add_popart_py_unit_test(synthetic_data_test VARIANTS Hw)
add_popart_py_unit_test(test_groupHostSync VARIANTS IpuModel2)
add_popart_py_unit_test(train_then_infer_test VARIANTS IpuModel2)
add_popart_py_unit_test(variable_inference_test)
add_popart_py_unit_test(virtual_graph_check_test)
add_popart_py_unit_test(virtual_graph_test VARIANTS IpuModel2)
add_popart_py_unit_test(type_casting_test)
add_popart_py_unit_test(supported_ops_test)
add_popart_py_unit_test(constant_op_test)
add_popart_py_unit_test(nop_test)
add_popart_py_unit_test(external_tensorproto_data_test)
add_popart_py_unit_test(debug_info_test)
add_popart_py_unit_test(logging_test)
add_popart_py_unit_test(model_loading_test)
add_popart_py_unit_test(int8_host_stream_test VARIANTS IpuModel2)
add_popart_py_unit_test(version_test)
add_popart_py_unit_test(test_all_reduce_op VARIANTS Hw)
add_popart_py_unit_test(min_max_unwinding_test)
add_popart_py_unit_test(deprecated_session_options)
add_popart_py_unit_test(constant_weights_error_test)

add_popart_py_unit_test(auto_loss_scaling_graph_replication_test VARIANTS Hw)
add_popart_py_unit_test(loss_grad_scaling_with_replication VARIANTS Hw)

add_unit_test(prng_test prng_test.cpp VARIANTS "${HW_EXCLUDE_C600}")
add_popart_py_unit_test(collectives_test VARIANTS "${HW_EXCLUDE_C600}")
add_popart_py_unit_test(dropout_replicated_pipeline VARIANTS "${HW_EXCLUDE_C600}")
add_popart_py_unit_test(graph_replication_test VARIANTS "${HW_EXCLUDE_C600}")
add_popart_py_unit_test(identical_replica_test VARIANTS "${HW_EXCLUDE_C600}")


# Test to check for boost in popart headers.
set(POPART_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/willow/include)
add_test(NAME "boost_free_interface_test"
  COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/boost_free_interface_test.py ${POPART_INCLUDE_DIR})

add_subdirectory(anchor_tests)
add_subdirectory(auto_virtual_graph_tests)
add_subdirectory(codelet_tests)
add_subdirectory(constexpr_tests)
add_subdirectory(dot_tests)
add_subdirectory(dropout_tests)
add_subdirectory(inplaceaccumulategradpartialsintooptimizeraccumtensor_tests)
add_subdirectory(inplace_tests)
add_subdirectory(lambserialisedweight_tests)
add_subdirectory(logical_if_tests)
add_subdirectory(matmul_tests)
add_subdirectory(operators_test)
add_subdirectory(optimizer_tests)
add_subdirectory(pattern_tests)
add_subdirectory(streamingmemory_tests)
add_subdirectory(streammode_tests)
add_subdirectory(pipelining_tests)
add_subdirectory(popart._internal.ir_cpp)
add_subdirectory(popxl)
add_subdirectory(recompute_tests)
add_subdirectory(scheduling_tests)
add_subdirectory(session_api_tests)
add_subdirectory(stepio_tests)
add_subdirectory(subgraph_tests)
add_subdirectory(tiedgather_tests)
add_subdirectory(topk_tests)
add_subdirectory(transformation_tests)
add_subdirectory(slice_pad_dual_tests)
add_subdirectory(unwinding_tests)
add_subdirectory(verify_cxx_11_interface)
add_subdirectory(customop_attr_pylist_tests)