6 #include "../ProfilingConnectionDumpToFileDecorator.hpp" 14 #include <boost/numeric/conversion/cast.hpp> 15 #include <boost/test/unit_test.hpp> 22 const std::vector<char> g_Data = {
'd',
'u',
'm',
'm',
'y' };
24 const unsigned char* g_DataPtr =
reinterpret_cast<const unsigned char*
>(g_Data.data());
29 DummyProfilingConnection()
31 , m_PacketData(
std::make_unique<unsigned char[]>(g_DataLength))
34 std::memcpy(m_PacketData.get(), g_DataPtr, g_DataLength);
35 m_Packet = std::make_unique<arm::pipe::Packet>(0u, g_DataLength, m_PacketData);
38 ~DummyProfilingConnection() =
default;
40 bool IsOpen()
const override 50 bool WritePacket(
const unsigned char* buffer, uint32_t length)
override 57 arm::pipe::Packet ReadPacket(uint32_t timeout)
override 60 return std::move(*m_Packet);
65 std::unique_ptr<unsigned char[]> m_PacketData;
66 std::unique_ptr<arm::pipe::Packet> m_Packet;
69 std::vector<char> ReadDumpFile(
const std::string& dumpFileName)
71 std::ifstream input(dumpFileName, std::ios::binary);
72 return std::vector<char>(std::istreambuf_iterator<char>(input), {});
81 armnn::Runtime::CreationOptions::ExternalProfilingOptions options;
82 options.m_IncomingCaptureFile =
"/";
83 options.m_OutgoingCaptureFile =
"";
90 armnn::Runtime::CreationOptions::ExternalProfilingOptions options;
91 options.m_IncomingCaptureFile =
"/";
92 options.m_OutgoingCaptureFile =
"";
101 armnn::Runtime::CreationOptions::ExternalProfilingOptions options;
102 options.m_IncomingCaptureFile = fileName.string();
103 options.m_OutgoingCaptureFile =
"";
108 std::unique_ptr<arm::pipe::Packet> packet;
109 BOOST_CHECK_NO_THROW(packet = std::make_unique<arm::pipe::Packet>(decorator.
ReadPacket(0)));
113 std::vector<char> data = ReadDumpFile(options.m_IncomingCaptureFile);
114 const char* packetData =
reinterpret_cast<const char*
>(packet->GetData());
117 constexpr
unsigned int bytesToSkip = 2u *
sizeof(uint32_t);
118 int diff = std::strncmp(data.data() + bytesToSkip, packetData, g_DataLength);
119 BOOST_CHECK(diff == 0);
120 fs::remove(fileName);
125 armnn::Runtime::CreationOptions::ExternalProfilingOptions options;
126 options.m_IncomingCaptureFile =
"";
127 options.m_OutgoingCaptureFile =
"/";
134 armnn::Runtime::CreationOptions::ExternalProfilingOptions options;
135 options.m_IncomingCaptureFile =
"";
136 options.m_OutgoingCaptureFile =
"/";
139 BOOST_CHECK_NO_THROW(decorator.
WritePacket(g_DataPtr, g_DataLength));
141 bool success = decorator.
WritePacket(g_DataPtr, g_DataLength);
142 BOOST_CHECK(!success);
149 armnn::Runtime::CreationOptions::ExternalProfilingOptions options;
150 options.m_IncomingCaptureFile =
"";
151 options.m_OutgoingCaptureFile = fileName.string();
155 bool success =
false;
156 BOOST_CHECK_NO_THROW(success = decorator.
WritePacket(g_DataPtr, g_DataLength));
157 BOOST_CHECK(success);
161 std::vector<char> data = ReadDumpFile(options.m_OutgoingCaptureFile);
164 int diff = std::strncmp(data.data(), g_Data.data(), g_DataLength);
165 BOOST_CHECK(diff == 0);
166 fs::remove(fileName);
bool WritePacket(const unsigned char *buffer, uint32_t length) override
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
void IgnoreUnused(Ts &&...)
arm::pipe::Packet ReadPacket(uint32_t timeout) override
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
BOOST_AUTO_TEST_CASE(CheckConvolution2dLayer)
BOOST_AUTO_TEST_SUITE_END()
fs::path NamedTempFile(const char *fileName)
Construct a temporary file name.
DataLayout::NCHW DataLayout::NCHW DataLayout::NHWC DataLayout::NHWC true