From 905befad57e78e710291d472a2970a118ac0d328 Mon Sep 17 00:00:00 2001 From: Kirill Golikov Date: Fri, 10 May 2024 23:48:04 +0200 Subject: [PATCH] [NeoML] reset PeakMemory in DnnBlobTest (#1067) Signed-off-by: Kirill Golikov --- NeoML/test/src/DnnBlobTest.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/NeoML/test/src/DnnBlobTest.cpp b/NeoML/test/src/DnnBlobTest.cpp index 12336734f..f93ed9fe7 100644 --- a/NeoML/test/src/DnnBlobTest.cpp +++ b/NeoML/test/src/DnnBlobTest.cpp @@ -26,14 +26,15 @@ using namespace NeoMLTest; TEST( CDnnBlobTest, InitWindowBlob ) { MathEngine().CleanUp(); + MathEngine().ResetPeakMemoryUsage(); { CPtr parent = CDnnBlob::CreateDataBlob( MathEngine(), CT_Float, 16, 1, 1 ); CPtr window = CDnnBlob::CreateWindowBlob( parent ); EXPECT_TRUE( window->GetData().IsNull() == false ); } - EXPECT_TRUE( MathEngine().GetPeakMemoryUsage() == 16 * sizeof( float ) ); EXPECT_TRUE( MathEngine().GetCurrentMemoryUsage() == 0 ); + EXPECT_EQ( MathEngine().GetPeakMemoryUsage(), 16 * sizeof( float ) ); } TEST( CDnnBlobTest, BufferTest )