From 53e06599a3af44db90c37d1cda34fc85ec9c27fa Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Wed, 5 Jan 2022 16:04:08 +0000 Subject: IVGCVSW-6672 Implement CanBeImported function to RefTensorHandle Signed-off-by: Nikhil Raj Change-Id: Icaa3aa7ef3e5cc3984941d095edfe8f0b2137879 --- .../reference/test/RefTensorHandleTests.cpp | 33 ++++++++++++++++++++++ 1 file changed, 33 insertions(+) (limited to 'src/backends/reference/test/RefTensorHandleTests.cpp') diff --git a/src/backends/reference/test/RefTensorHandleTests.cpp b/src/backends/reference/test/RefTensorHandleTests.cpp index 39f5a2aeed..3504f53bc7 100644 --- a/src/backends/reference/test/RefTensorHandleTests.cpp +++ b/src/backends/reference/test/RefTensorHandleTests.cpp @@ -253,6 +253,39 @@ TEST_CASE("MisalignedPointer") delete[] testPtr; } +TEST_CASE("CheckCanBeImported") +{ + TensorInfo info({1}, DataType::Float32); + RefTensorHandle handle(info, static_cast(MemorySource::Malloc)); + + int* testPtr = new int(4); + + // Not supported + CHECK(!handle.CanBeImported(static_cast(testPtr), MemorySource::DmaBuf)); + + // Supported + CHECK(handle.CanBeImported(static_cast(testPtr), MemorySource::Malloc)); + + delete testPtr; + +} + +TEST_CASE("MisalignedCanBeImported") +{ + TensorInfo info({2}, DataType::Float32); + RefTensorHandle handle(info, static_cast(MemorySource::Malloc)); + + // Allocate a 2 int array + int* testPtr = new int[2]; + + // Increment pointer by 1 byte + void* misalignedPtr = static_cast(reinterpret_cast(testPtr) + 1); + + CHECK(!handle.Import(misalignedPtr, MemorySource::Malloc)); + + delete[] testPtr; +} + #endif } -- cgit v1.2.1