1 files changed, 22 insertions, 0 deletions
diff --git a/src/backends/neon/test/NeonTensorHandleTests.cpp b/src/backends/neon/test/NeonTensorHandleTests.cpp
index 9d69a5c2d0..c8e781b71d 100644
--- a/src/backends/neon/test/NeonTensorHandleTests.cpp
+++ b/src/backends/neon/test/NeonTensorHandleTests.cpp
@@ -703,6 +703,28 @@ TEST_CASE("NeonTensorHandleFactoryImport")
     memoryManager->Release();
 }
 
+TEST_CASE("NeonTensorHandleCanBeImported")
+{
+    std::shared_ptr<NeonMemoryManager> memoryManager = std::make_shared<NeonMemoryManager>(
+        std::make_unique<arm_compute::Allocator>(),
+        BaseMemoryManager::MemoryAffinity::Offset);
+    NeonTensorHandleFactory handleFactory(memoryManager);
+    TensorInfo info({ 1, 1, 2, 1 }, DataType::Float32);
+
+    // create TensorHandle (Memory Managed status is irrelevant)
+    auto handle = handleFactory.CreateTensorHandle(info, false);
+
+    // Create an aligned buffer
+    float alignedBuffer[2] = { 2.5f, 5.5f };
+    // Check aligned buffers return true
+    CHECK(handle->CanBeImported(&alignedBuffer, MemorySource::Malloc) == true);
+
+    // Create a misaligned buffer from the aligned one
+    float* misalignedBuffer = reinterpret_cast<float*>(reinterpret_cast<char*>(alignedBuffer) + 1);
+    // Check misaligned buffers return false
+    CHECK(handle->CanBeImported(static_cast<void*>(misalignedBuffer), MemorySource::Malloc) == false);
+}
+
 TEST_CASE("NeonTensorHandleSupportsInPlaceComputation")
 {
     std::shared_ptr<NeonMemoryManager> memoryManager = std::make_shared<NeonMemoryManager>();