arrayfire/test/memory_lock.cpp at master · arrayfire/arrayfire · GitHub

History

73 lines (57 loc) · 1.94 KB

Raw

/*******************************************************

* This file is distributed under 3-clause BSD license.

* The complete license agreement can be obtained at:

* http://arrayfire.com/licenses/BSD-3-Clause

********************************************************/

#include <arrayfire.h>

#include <gtest/gtest.h>

#include <testHelpers.hpp>

#include <af/dim4.hpp>

#include <af/traits.hpp>

#include <iostream>

#include <string>

#include <vector>

using af::array;

using af::deviceGC;

using af::deviceMemInfo;

using std::cout;

using std::endl;

using std::string;

using std::vector;

const size_t step_bytes = 1024;

// This test should be by itself as it leaks memory intentionally

TEST(Memory, lock) {

size_t alloc_bytes, alloc_buffers;

size_t lock_bytes, lock_buffers;

cleanSlate(); // Clean up everything done so far

const dim_t num = step_bytes / sizeof(float);

vector<float> in(num);

af_array arr = 0;

ASSERT_SUCCESS(af_create_array(&arr, &in[0], 1, &num, f32));

deviceMemInfo(&alloc_bytes, &alloc_buffers, &lock_bytes, &lock_buffers);

ASSERT_EQ(alloc_buffers, 1u);

ASSERT_EQ(lock_buffers, 1u);

ASSERT_EQ(alloc_bytes, step_bytes);

ASSERT_EQ(lock_bytes, step_bytes);

// arr1 gets released by end of the following code block

{

array a(arr);

a.lock();

// No new memory should be allocated

deviceMemInfo(&alloc_bytes, &alloc_buffers, &lock_bytes, &lock_buffers);

ASSERT_EQ(alloc_buffers, 1u);

ASSERT_EQ(lock_buffers, 1u);

ASSERT_EQ(alloc_bytes, step_bytes);

ASSERT_EQ(lock_bytes, step_bytes);

}

// Making sure all unlocked buffers are freed

deviceGC();

deviceMemInfo(&alloc_bytes, &alloc_buffers, &lock_bytes, &lock_buffers);

ASSERT_EQ(alloc_buffers, 1u);

ASSERT_EQ(lock_buffers, 1u);

ASSERT_EQ(alloc_bytes, step_bytes);

ASSERT_EQ(lock_bytes, step_bytes);

}

arrayfire/test/memory_lock.cpp at master · arrayfire/arrayfire · GitHub

Uh oh!

Expand file tree

Breadcrumbs

memory_lock.cpp

Latest commit

History

Breadcrumbs

memory_lock.cpp

File metadata and controls