[NeoML] reset PeakMemory in DnnBlobTest (#1067)

Signed-off-by: Kirill Golikov <kirill.golikov@abbyy.com>
neoml-lib · May 10, 2024 · 905befa · 905befa
1 parent daf71b4
commit 905befa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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<CDnnBlob> parent = CDnnBlob::CreateDataBlob( MathEngine(), CT_Float, 16, 1, 1 );
         CPtr<CDnnBlob> 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 )