Context creation flags in AubDump

Change-Id: Ia9651d37fa716351728633f0bee5b0eae98e758f

Author: pwilma

runtime/aub_mem_dump/CMakeLists.txt

@@ -12,6 +12,7 @@ set(RUNTIME_SRCS_AUB_MEM_DUMP
   ${CMAKE_CURRENT_SOURCE_DIR}/aub_mem_dump.h
   ${CMAKE_CURRENT_SOURCE_DIR}/aub_mem_dump.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/aub_services.h
+  ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/context_flags.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/page_table_entry_bits.h
 )
 target_sources(${NEO_STATIC_LIB_NAME} PRIVATE ${RUNTIME_SRCS_AUB_MEM_DUMP})

runtime/aub_mem_dump/aub_mem_dump.cpp

@@ -137,9 +137,11 @@ void LrcaHelper::initialize(void *pLRCIn) const {
     auto pLRI = ptrOffset(pLRCA, offsetLRI0);
     auto numRegs = numRegsLRI0;
     *pLRI++ = 0x11001000 | (2 * numRegs - 1);
+    uint32_t ctxSrCtlValue = 0x00010001; // Inhibit context-restore
+    setContextSaveRestoreFlags(ctxSrCtlValue);
     while (numRegs-- > 0) {
         *pLRI++ = mmioBase + 0x2244; // CTXT_SR_CTL
-        *pLRI++ = 0x00010001;        // Inhibit context-restore
+        *pLRI++ = ctxSrCtlValue;
     }
 
     // Initialize the other LRI

runtime/aub_mem_dump/aub_mem_dump.h

@@ -343,6 +343,7 @@ struct LrcaHelper {
     void setPDP3(void *pLRCIn, uint64_t address) const;
 
     void setPML4(void *pLRCIn, uint64_t address) const;
+    MOCKABLE_VIRTUAL void setContextSaveRestoreFlags(uint32_t &value) const;
 };
 
 struct LrcaHelperRcs : public LrcaHelper {

runtime/aub_mem_dump/context_flags.cpp

@@ -0,0 +1,13 @@
+/*
+ * Copyright (C) 2017-2018 Intel Corporation
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ */
+
+#include "runtime/aub_mem_dump/aub_mem_dump.h"
+
+namespace AubMemDump {
+void LrcaHelper::setContextSaveRestoreFlags(uint32_t &ctxSrCtlValue) const {
+}
+} // namespace AubMemDump

unit_tests/aub/aub_helper_tests.cpp

@@ -9,6 +9,7 @@
 #include "runtime/aub/aub_helper.h"
 #include "runtime/aub_mem_dump/aub_mem_dump.h"
 #include "runtime/aub_mem_dump/page_table_entry_bits.h"
+#include "runtime/command_stream/aub_command_stream_receiver_hw.h"
 #include "unit_tests/fixtures/device_fixture.h"
 #include "test.h"
 
@@ -100,3 +101,22 @@ HWTEST_F(AubHelperHwTest, GivenEnabledLocalMemoryWhenGetMemTraceForPtEntryIsCall
     int addressSpace = aubHelper.getMemTraceForPtEntry();
     EXPECT_EQ(AubMemDump::AddressSpaceValues::TraceLocal, addressSpace);
 }
+
+struct MockLrcaHelper : AubMemDump::LrcaHelper {
+    mutable uint32_t setContextSaveRestoreFlagsCalled = 0;
+    MockLrcaHelper(uint32_t base) : AubMemDump::LrcaHelper(base) {}
+    void setContextSaveRestoreFlags(uint32_t &value) const override {
+        setContextSaveRestoreFlagsCalled++;
+        AubMemDump::LrcaHelper::setContextSaveRestoreFlags(value);
+    }
+};
+
+HWTEST_F(AubHelperHwTest, giverLrcaHelperWhenContextIsInitializedThenContextFlagsAreSet) {
+    const auto &csTraits = AUBCommandStreamReceiverHw<FamilyType>::getCsTraits(EngineType::ENGINE_RCS);
+    MockLrcaHelper lrcaHelper(csTraits.mmioBase);
+
+    std::unique_ptr<void, std::function<void(void *)>> lrcaBase(alignedMalloc(csTraits.sizeLRCA, csTraits.alignLRCA), alignedFree);
+
+    lrcaHelper.initialize(lrcaBase.get());
+    ASSERT_NE(0u, lrcaHelper.setContextSaveRestoreFlagsCalled);
+}