| 808 | af_memory_manager manager; |
| 809 | std::unique_ptr<E2ETestPayload> payload{new E2ETestPayload()}; |
| 810 | void SetUp() override { |
| 811 | af_create_memory_manager(&manager); |
| 812 | |
| 813 | // Set payload_fn |
| 814 | af_memory_manager_set_payload(manager, payload.get()); |
| 815 | |
| 816 | auto initialize_fn = [](af_memory_manager manager) { |
| 817 | auto *payload = getMemoryManagerPayload<E2ETestPayload>(manager); |
| 818 | payload->initializeCalledTimes++; |
| 819 | return AF_SUCCESS; |
| 820 | }; |
| 821 | af_memory_manager_set_initialize_fn(manager, initialize_fn); |
| 822 | |
| 823 | auto shutdown_fn = [](af_memory_manager manager) { |
| 824 | auto *payload = getMemoryManagerPayload<E2ETestPayload>(manager); |
| 825 | payload->shutdownCalledTimes++; |
| 826 | return AF_SUCCESS; |
| 827 | }; |
| 828 | af_memory_manager_set_shutdown_fn(manager, shutdown_fn); |
| 829 | |
| 830 | // alloc |
| 831 | af_memory_manager_set_alloc_fn(manager, alloc_fn); |
| 832 | af_memory_manager_set_allocated_fn(manager, allocated_fn); |
| 833 | af_memory_manager_set_unlock_fn(manager, unlock_fn); |
| 834 | // utils |
| 835 | af_memory_manager_set_signal_memory_cleanup_fn( |
| 836 | manager, signal_memory_cleanup_fn); |
| 837 | af_memory_manager_set_print_info_fn(manager, print_info_fn); |
| 838 | // user lock/unlock |
| 839 | af_memory_manager_set_user_lock_fn(manager, user_lock_fn); |
| 840 | af_memory_manager_set_user_unlock_fn(manager, user_unlock_fn); |
| 841 | af_memory_manager_set_is_user_locked_fn(manager, is_user_locked_fn); |
| 842 | // memory pressure |
| 843 | af_memory_manager_set_get_memory_pressure_fn(manager, |
| 844 | get_memory_pressure_fn); |
| 845 | af_memory_manager_set_jit_tree_exceeds_memory_pressure_fn( |
| 846 | manager, jit_tree_exceeds_memory_pressure_fn); |
| 847 | // ocl |
| 848 | af_memory_manager_set_add_memory_management_fn( |
| 849 | manager, add_memory_management_fn); |
| 850 | af_memory_manager_set_remove_memory_management_fn( |
| 851 | manager, remove_memory_management_fn); |
| 852 | |
| 853 | af_set_memory_manager(manager); |
| 854 | } |
| 855 | |
| 856 | void TearDown() override { |
| 857 | af_device_gc(); |
nothing calls this directly
no test coverage detected