diff options
Diffstat (limited to 'test/TestTensor.hpp')
-rw-r--r-- | test/TestTensor.hpp | 4 |
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; |