aboutsummaryrefslogtreecommitdiff
path: root/test/TestTensor.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'test/TestTensor.hpp')
-rw-r--r--test/TestTensor.hpp4
1 files changed, 3 insertions, 1 deletions
diff --git a/test/TestTensor.hpp b/test/TestTensor.hpp
index 623c9fbd..1cd1950d 100644
--- a/test/TestTensor.hpp
+++ b/test/TestTensor.hpp
@@ -6,6 +6,8 @@
#include "../ArmnnDriver.hpp"
+#include <armnn/utility/Assert.hpp>
+
namespace driverTestHelpers
{
@@ -17,7 +19,7 @@ public:
: m_Shape{shape}
, m_Data{data}
{
- BOOST_ASSERT(m_Shape.GetNumElements() == m_Data.size());
+ ARMNN_ASSERT(m_Shape.GetNumElements() == m_Data.size());
}
hidl_vec<uint32_t> GetDimensions() const;