ClImportTensorHandleFactoryTests.cpp File Reference

#include <cl/ClImportTensorHandleFactory.hpp>
#include <boost/test/unit_test.hpp>

Go to the source code of this file.

Functions
	BOOST_AUTO_TEST_CASE (ImportTensorFactoryAskedToCreateManagedTensorThrowsException)
	BOOST_AUTO_TEST_CASE (ImportTensorFactoryCreateMallocTensorHandle)
	BOOST_AUTO_TEST_CASE (CreateSubtensorOfImportTensor)
	BOOST_AUTO_TEST_CASE (CreateSubtensorNonZeroXYIsInvalid)
	BOOST_AUTO_TEST_CASE (CreateSubtensorXYMustMatchParent)
	BOOST_AUTO_TEST_CASE (CreateSubtensorMustBeSmallerThanParent)

Function Documentation

BOOST_AUTO_TEST_CASE() [1/6]

BOOST_AUTO_TEST_CASE

(

ImportTensorFactoryAskedToCreateManagedTensorThrowsException

)

Definition at line 13 of file ClImportTensorHandleFactoryTests.cpp.

References ClImportTensorHandleFactory::CreateTensorHandle(), armnn::Malloc, and armnn::NCHW.

{
    // Create the factory to import tensors.
    ClImportTensorHandleFactory factory(static_cast<MemorySourceFlags>(MemorySource::Malloc),
                                        static_cast<MemorySourceFlags>(MemorySource::Malloc));
    TensorInfo tensorInfo;
    // This factory is designed to import the memory of tensors. Asking for a handle that requires
    // a memory manager should result in an exception.
    BOOST_REQUIRE_THROW(factory.CreateTensorHandle(tensorInfo, true), InvalidArgumentException);
    BOOST_REQUIRE_THROW(factory.CreateTensorHandle(tensorInfo, DataLayout::NCHW, true), InvalidArgumentException);
}

BOOST_AUTO_TEST_CASE() [2/6]

BOOST_AUTO_TEST_CASE

(

ImportTensorFactoryCreateMallocTensorHandle

)

Definition at line 25 of file ClImportTensorHandleFactoryTests.cpp.

References ClImportTensorHandleFactory::CreateTensorHandle(), armnn::Float32, armnn::Malloc, and armnn::NHWC.

{
    // Create the factory to import tensors.
    ClImportTensorHandleFactory factory(static_cast<MemorySourceFlags>(MemorySource::Malloc),
                                        static_cast<MemorySourceFlags>(MemorySource::Malloc));
    TensorShape tensorShape{ 6, 7, 8, 9 };
    TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
    // Start with the TensorInfo factory method. Create an import tensor handle and verify the data is
    // passed through correctly.
    auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
    BOOST_ASSERT(tensorHandle);
    BOOST_ASSERT(tensorHandle->GetImportFlags() == static_cast<MemorySourceFlags>(MemorySource::Malloc));
    BOOST_ASSERT(tensorHandle->GetShape() == tensorShape);

    // Same method but explicitly specifying isManaged = false.
    tensorHandle = factory.CreateTensorHandle(tensorInfo, false);
    BOOST_CHECK(tensorHandle);
    BOOST_ASSERT(tensorHandle->GetImportFlags() == static_cast<MemorySourceFlags>(MemorySource::Malloc));
    BOOST_ASSERT(tensorHandle->GetShape() == tensorShape);

    // Now try TensorInfo and DataLayout factory method.
    tensorHandle = factory.CreateTensorHandle(tensorInfo, DataLayout::NHWC);
    BOOST_CHECK(tensorHandle);
    BOOST_ASSERT(tensorHandle->GetImportFlags() == static_cast<MemorySourceFlags>(MemorySource::Malloc));
    BOOST_ASSERT(tensorHandle->GetShape() == tensorShape);
}

BOOST_AUTO_TEST_CASE() [3/6]

BOOST_AUTO_TEST_CASE

(

CreateSubtensorOfImportTensor

)

Definition at line 52 of file ClImportTensorHandleFactoryTests.cpp.

References ClImportTensorHandleFactory::CreateSubTensorHandle(), ClImportTensorHandleFactory::CreateTensorHandle(), armnn::Float32, and armnn::Malloc.

{
    // Create the factory to import tensors.
    ClImportTensorHandleFactory factory(static_cast<MemorySourceFlags>(MemorySource::Malloc),
                                        static_cast<MemorySourceFlags>(MemorySource::Malloc));
    // Create a standard inport tensor.
    TensorShape tensorShape{ 224, 224, 1, 1 };
    TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
    auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
    // Use the factory to create a 16x16 sub tensor.
    TensorShape subTensorShape{ 16, 16, 1, 1 };
    // Starting at an offset of 1x1.
    uint32_t origin[4] = { 1, 1, 0, 0 };
    auto subTensor     = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
    BOOST_CHECK(subTensor);
    BOOST_ASSERT(subTensor->GetShape() == subTensorShape);
    BOOST_ASSERT(subTensor->GetParent() == tensorHandle.get());
}

BOOST_AUTO_TEST_CASE() [4/6]

BOOST_AUTO_TEST_CASE

(

CreateSubtensorNonZeroXYIsInvalid

)

Definition at line 71 of file ClImportTensorHandleFactoryTests.cpp.

References ClImportTensorHandleFactory::CreateSubTensorHandle(), ClImportTensorHandleFactory::CreateTensorHandle(), armnn::Float32, and armnn::Malloc.

{
    // Create the factory to import tensors.
    ClImportTensorHandleFactory factory(static_cast<MemorySourceFlags>(MemorySource::Malloc),
                                        static_cast<MemorySourceFlags>(MemorySource::Malloc));
    // Create a standard import tensor.
    TensorShape tensorShape{ 224, 224, 1, 1 };
    TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
    auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
    // Use the factory to create a 16x16 sub tensor.
    TensorShape subTensorShape{ 16, 16, 1, 1 };
    // This looks a bit backwards because of how Cl specifies tensors. Essentially we want to trigger our
    // check "(coords.x() != 0 || coords.y() != 0)"
    uint32_t origin[4] = { 0, 0, 1, 1 };
    auto subTensor     = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
    // We expect a nullptr.
    BOOST_ASSERT(subTensor == nullptr);
}

BOOST_AUTO_TEST_CASE() [5/6]

BOOST_AUTO_TEST_CASE

(

CreateSubtensorXYMustMatchParent

)

Definition at line 90 of file ClImportTensorHandleFactoryTests.cpp.

References ClImportTensorHandleFactory::CreateSubTensorHandle(), ClImportTensorHandleFactory::CreateTensorHandle(), armnn::Float32, and armnn::Malloc.

{
    // Create the factory to import tensors.
    ClImportTensorHandleFactory factory(static_cast<MemorySourceFlags>(MemorySource::Malloc),
                                        static_cast<MemorySourceFlags>(MemorySource::Malloc));
    // Create a standard import tensor.
    TensorShape tensorShape{ 224, 224, 1, 1 };
    TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
    auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
    // Use the factory to create a 16x16 sub tensor but make the CL x and y axis different.
    TensorShape subTensorShape{ 16, 16, 2, 2 };
    // We want to trigger our ((parentShape.x() != shape.x()) || (parentShape.y() != shape.y()))
    uint32_t origin[4] = { 1, 1, 0, 0 };
    auto subTensor     = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
    // We expect a nullptr.
    BOOST_ASSERT(subTensor == nullptr);
}

BOOST_AUTO_TEST_CASE() [6/6]

BOOST_AUTO_TEST_CASE

(

CreateSubtensorMustBeSmallerThanParent

)

Definition at line 108 of file ClImportTensorHandleFactoryTests.cpp.

References BOOST_AUTO_TEST_SUITE_END(), ClImportTensorHandleFactory::CreateSubTensorHandle(), ClImportTensorHandleFactory::CreateTensorHandle(), armnn::Float32, and armnn::Malloc.

{
    // Create the factory to import tensors.
    ClImportTensorHandleFactory factory(static_cast<MemorySourceFlags>(MemorySource::Malloc),
                                        static_cast<MemorySourceFlags>(MemorySource::Malloc));
    // Create a standard import tensor.
    TensorShape tensorShape{ 224, 224, 1, 1 };
    TensorInfo tensorInfo(tensorShape, armnn::DataType::Float32);
    auto tensorHandle = factory.CreateTensorHandle(tensorInfo);
    // Ask for a subtensor that's the same size as the parent.
    TensorShape subTensorShape{ 224, 224, 1, 1 };
    uint32_t origin[4] = { 1, 1, 0, 0 };
    // This should result in a nullptr.
    auto subTensor = factory.CreateSubTensorHandle(*tensorHandle, subTensorShape, origin);
    BOOST_ASSERT(subTensor == nullptr);
}