Function TEST_CASE

tests/validation/NEON/UNIT/MemoryManager.cpp:49–91

TEST_SUITE(MemoryManager)

TEST_CASE(BlobMemoryManagerSimpleWithinFunctionLevel, framework::DatasetMode::ALL)
{
    Allocator allocator{};
    auto      lifetime_mgr = std::make_shared<BlobLifetimeManager>();
    auto      pool_mgr     = std::make_shared<PoolManager>();
    auto      mm           = std::make_shared<MemoryManagerOnDemand>(lifetime_mgr, pool_mgr);

    // Create tensors
    Tensor src = create_tensor<Tensor>(TensorShape(27U, 11U, 3U), DataType::F32, 1);
    Tensor dst = create_tensor<Tensor>(TensorShape(27U, 11U, 3U), DataType::F32, 1);

    // Create and configure function
    NENormalizationLayer norm_layer_1(mm);
    NENormalizationLayer norm_layer_2(mm);
    norm_layer_1.configure(&src, &dst, NormalizationLayerInfo(NormType::CROSS_MAP, 3));
    norm_layer_2.configure(&src, &dst, NormalizationLayerInfo(NormType::IN_MAP_1D, 3));

    ARM_COMPUTE_ASSERT(src.info()->is_resizable());
    ARM_COMPUTE_ASSERT(dst.info()->is_resizable());

    // Allocate tensors
    src.allocator()->allocate();
    dst.allocator()->allocate();

    ARM_COMPUTE_ASSERT(!src.info()->is_resizable());
    ARM_COMPUTE_ASSERT(!dst.info()->is_resizable());

    // Finalize memory manager
    mm->populate(allocator, 1 /* num_pools */);
    ARM_COMPUTE_EXPECT(mm->pool_manager()->num_pools() == 1, framework::LogLevel::ERRORS);
    ARM_COMPUTE_EXPECT(mm->lifetime_manager()->are_all_finalized(), framework::LogLevel::ERRORS);

    // Fill tensors
    arm_compute::test::library->fill_tensor_uniform(Accessor(src), 0);

    // Compute functions
    norm_layer_1.run();
    norm_layer_2.run();

    // Clear manager
    mm->clear();
    ARM_COMPUTE_EXPECT(mm->pool_manager()->num_pools() == 0, framework::LogLevel::ERRORS);
}

TEST_SUITE_END()
TEST_SUITE_END()