From 52e48c4215b2f8741537780dd8bf1ff863efdc7f Mon Sep 17 00:00:00 2001
From: Mateusz Hoppe <mateusz.hoppe@intel.com>
Date: Fri, 5 Apr 2019 13:05:06 +0200
Subject: [PATCH] Add --enable_memory_dumps option to aub_tests

Related-To: NEO-3079

Change-Id: I185309aba2a8a6e02988bbe407c43ea7fa53be4d
Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
---
 unit_tests/main.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/unit_tests/main.cpp b/unit_tests/main.cpp
index 82cbe167f6..a21ce9dd9b 100644
--- a/unit_tests/main.cpp
+++ b/unit_tests/main.cpp
@@ -174,6 +174,7 @@ int main(int argc, char **argv) {
     bool useDefaultListener = false;
     bool enable_alarm = true;
     bool setupFeatureTable = testMode == TestMode::AubTests ? true : false;
+    bool enableMemoryDumps = false;
 
     applyWorkarounds();
 
@@ -285,6 +286,8 @@ int main(int argc, char **argv) {
                 DebugManager.setReaderImpl(SettingsReader::create());
                 DebugManager.injectSettingsFromReader();
             }
+        } else if (!strcmp("--enable_memory_dumps", argv[i]) && testMode == TestMode::AubTests) {
+            enableMemoryDumps = true;
         }
     }
 
@@ -292,6 +295,11 @@ int main(int argc, char **argv) {
         return -1;
     }
 
+    if (enableMemoryDumps) {
+        DebugManager.flags.AUBDumpBufferFormat.set("BIN");
+        DebugManager.flags.AUBDumpImageFormat.set("TRE");
+    }
+
     uint32_t threadsPerEu = hwInfoConfigFactory[productFamily]->threadsPerEu;
     PLATFORM platform;
     auto hardwareInfo = hardwareInfoTable[productFamily];