e87cc3fa9e
* Added OpenVINO Tensor API * Tensor API improvements * Moved Tensor to ngraph * Moved Tensor tests * Fixed docs and code style * Trying to fix Windows * Fixed clang-format * Moved Tensor to runtime namespace * Fixed compilation * Fixed clang-format * Fixed tests in debug Co-authored-by: apankratovantonp <anton.pankratov@intel.com>
60 lines
1.9 KiB
C++
60 lines
1.9 KiB
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <memory>
|
|
|
|
#include "openvino/core/except.hpp"
|
|
#include "openvino/runtime/allocator.hpp"
|
|
|
|
using OVDefaultAllocatorTest = ::testing::Test;
|
|
|
|
TEST_F(OVDefaultAllocatorTest, notThrowOnZeroSize) {
|
|
ov::runtime::Allocator allocator;
|
|
void* ptr = nullptr;
|
|
ASSERT_NO_THROW(ptr = allocator.allocate(0));
|
|
ASSERT_NO_THROW(allocator.deallocate(ptr));
|
|
}
|
|
|
|
TEST_F(OVDefaultAllocatorTest, canAllocateAndDeallocate) {
|
|
ov::runtime::Allocator allocator;
|
|
void* ptr = nullptr;
|
|
ASSERT_NO_THROW(ptr = allocator.allocate(64));
|
|
ASSERT_NO_THROW(allocator.deallocate(ptr));
|
|
}
|
|
|
|
TEST_F(OVDefaultAllocatorTest, alignedAllocationIsNotImplemented) {
|
|
ov::runtime::Allocator allocator;
|
|
ASSERT_THROW(allocator.allocate(64, 64), ov::Exception);
|
|
}
|
|
|
|
TEST_F(OVDefaultAllocatorTest, sizedAndAlignedDeallocationAreNotImplemented) {
|
|
ov::runtime::Allocator allocator;
|
|
void* ptr = nullptr;
|
|
ASSERT_NO_THROW(ptr = allocator.allocate(64));
|
|
ASSERT_THROW(allocator.deallocate(ptr, 64), ov::Exception);
|
|
ASSERT_THROW(allocator.deallocate(ptr, 0, 64), ov::Exception);
|
|
ASSERT_NO_THROW(allocator.deallocate(ptr));
|
|
}
|
|
|
|
TEST_F(OVDefaultAllocatorTest, defaultAllocatorsAreEqual) {
|
|
ov::runtime::Allocator allocator0, allocator1;
|
|
ASSERT_TRUE(allocator0 == allocator1);
|
|
void* ptr = nullptr;
|
|
ASSERT_NO_THROW(ptr = allocator0.allocate(64));
|
|
ASSERT_NO_THROW(allocator1.deallocate(ptr));
|
|
ASSERT_NO_THROW(ptr = allocator1.allocate(64));
|
|
ASSERT_NO_THROW(allocator0.deallocate(ptr));
|
|
}
|
|
|
|
TEST_F(OVDefaultAllocatorTest, canAllocate10KMemory) {
|
|
ov::runtime::Allocator allocator;
|
|
// large block such as 10k will result in sigsegv if not allocated
|
|
void* handle = allocator.allocate(10000);
|
|
char* ptr = reinterpret_cast<char*>(handle);
|
|
ptr[9999] = 11;
|
|
EXPECT_EQ(ptr[9999], 11);
|
|
allocator.deallocate(handle);
|
|
} |