MCPcopy Create free account
hub / github.com/arrayfire/arrayfire / SetUp

Method SetUp

test/memory.cpp:810–854  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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();

Callers

nothing calls this directly

Tested by

no test coverage detected