AUB CSRs to use a shared address mapper (CPU VA to GTT VA)

This commit moves address mapper from CSR to execution environment
in order to generate unique GTT VA for LRCA, HWSP and ring buffer
between different CSRs.
Additionally, moved the rest of AUB file stream tests to separate module.

Change-Id: I02ae44202c0255277a7ac17532485419e0c403ab

Author: smilczar

runtime/command_stream/aub_center.h

@@ -7,13 +7,15 @@
 
 #pragma once
 #include "runtime/command_stream/aub_stream_provider.h"
+#include "runtime/memory_manager/address_mapper.h"
 #include "runtime/memory_manager/physical_address_allocator.h"
 
 namespace OCLRT {
 
 class AubCenter {
   public:
     AubCenter() {
+        addressMapper = std::make_unique<AddressMapper>();
         streamProvider = std::make_unique<AubFileStreamProvider>();
     }
     virtual ~AubCenter() = default;
@@ -26,12 +28,17 @@ class AubCenter {
         return physicalAddressAllocator.get();
     }
 
+    AddressMapper *getAddressMapper() const {
+        return addressMapper.get();
+    }
+
     AubStreamProvider *getStreamProvider() const {
         return streamProvider.get();
     }
 
   protected:
     std::unique_ptr<PhysicalAddressAllocator> physicalAddressAllocator;
+    std::unique_ptr<AddressMapper> addressMapper;
     std::unique_ptr<AubStreamProvider> streamProvider;
 };
 } // namespace OCLRT

runtime/command_stream/aub_command_stream_receiver_hw.h

@@ -62,7 +62,7 @@ class AUBCommandStreamReceiverHw : public CommandStreamReceiverSimulatedHw<GfxFa
     MOCKABLE_VIRTUAL bool reopenFile(const std::string &fileName);
     MOCKABLE_VIRTUAL void initFile(const std::string &fileName);
     MOCKABLE_VIRTUAL void closeFile();
-    MOCKABLE_VIRTUAL bool isFileOpen();
+    MOCKABLE_VIRTUAL bool isFileOpen() const;
     MOCKABLE_VIRTUAL const std::string &getFileName();
 
     void initializeEngine(EngineType engineType);
@@ -95,7 +95,7 @@ class AUBCommandStreamReceiverHw : public CommandStreamReceiverSimulatedHw<GfxFa
     std::unique_ptr<TypeSelector<PML4, PDPE, sizeof(void *) == 8>::type> ppgtt;
     std::unique_ptr<PDPE> ggtt;
     // remap CPU VA -> GGTT VA
-    AddressMapper gttRemap;
+    AddressMapper *gttRemap;
 
     MOCKABLE_VIRTUAL bool addPatchInfoComments();
     void addGUCStartMessage(uint64_t batchBufferAddress, EngineType engineType);

runtime/command_stream/aub_command_stream_receiver_hw.inl

@@ -46,6 +46,9 @@ AUBCommandStreamReceiverHw<GfxFamily>::AUBCommandStreamReceiverHw(const Hardware
     ppgtt = std::make_unique<TypeSelector<PML4, PDPE, sizeof(void *) == 8>::type>(physicalAddressAllocator);
     ggtt = std::make_unique<PDPE>(physicalAddressAllocator);
 
+    gttRemap = aubCenter->getAddressMapper();
+    UNRECOVERABLE_IF(nullptr == gttRemap);
+
     auto streamProvider = aubCenter->getStreamProvider();
     UNRECOVERABLE_IF(nullptr == streamProvider);
 
@@ -143,7 +146,7 @@ void AUBCommandStreamReceiverHw<GfxFamily>::closeFile() {
 }
 
 template <typename GfxFamily>
-bool AUBCommandStreamReceiverHw<GfxFamily>::isFileOpen() {
+bool AUBCommandStreamReceiverHw<GfxFamily>::isFileOpen() const {
     return stream->isOpen();
 }
 
@@ -165,7 +168,7 @@ void AUBCommandStreamReceiverHw<GfxFamily>::initializeEngine(EngineType engineTy
         const size_t sizeHWSP = 0x1000;
         const size_t alignHWSP = 0x1000;
         engineInfo.pGlobalHWStatusPage = alignedMalloc(sizeHWSP, alignHWSP);
-        engineInfo.ggttHWSP = gttRemap.map(engineInfo.pGlobalHWStatusPage, sizeHWSP);
+        engineInfo.ggttHWSP = gttRemap->map(engineInfo.pGlobalHWStatusPage, sizeHWSP);
 
         auto physHWSP = ggtt->map(engineInfo.ggttHWSP, sizeHWSP, this->getGTTBits(), getMemoryBankForGtt());
 
@@ -197,7 +200,7 @@ void AUBCommandStreamReceiverHw<GfxFamily>::initializeEngine(EngineType engineTy
     {
         const size_t alignRingBuffer = 0x1000;
         engineInfo.pRingBuffer = alignedMalloc(engineInfo.sizeRingBuffer, alignRingBuffer);
-        engineInfo.ggttRingBuffer = gttRemap.map(engineInfo.pRingBuffer, engineInfo.sizeRingBuffer);
+        engineInfo.ggttRingBuffer = gttRemap->map(engineInfo.pRingBuffer, engineInfo.sizeRingBuffer);
         auto physRingBuffer = ggtt->map(engineInfo.ggttRingBuffer, engineInfo.sizeRingBuffer, this->getGTTBits(), getMemoryBankForGtt());
 
         {
@@ -225,7 +228,7 @@ void AUBCommandStreamReceiverHw<GfxFamily>::initializeEngine(EngineType engineTy
 
     // Write our LRCA
     {
-        engineInfo.ggttLRCA = gttRemap.map(engineInfo.pLRCA, sizeLRCA);
+        engineInfo.ggttLRCA = gttRemap->map(engineInfo.pLRCA, sizeLRCA);
         auto lrcAddressPhys = ggtt->map(engineInfo.ggttLRCA, sizeLRCA, this->getGTTBits(), getMemoryBankForGtt());
 
         {
@@ -254,15 +257,15 @@ template <typename GfxFamily>
 void AUBCommandStreamReceiverHw<GfxFamily>::freeEngineInfoTable() {
     for (auto &engineInfo : engineInfoTable) {
         alignedFree(engineInfo.pLRCA);
-        gttRemap.unmap(engineInfo.pLRCA);
+        gttRemap->unmap(engineInfo.pLRCA);
         engineInfo.pLRCA = nullptr;
 
         alignedFree(engineInfo.pGlobalHWStatusPage);
-        gttRemap.unmap(engineInfo.pGlobalHWStatusPage);
+        gttRemap->unmap(engineInfo.pGlobalHWStatusPage);
         engineInfo.pGlobalHWStatusPage = nullptr;
 
         alignedFree(engineInfo.pRingBuffer);
-        gttRemap.unmap(engineInfo.pRingBuffer);
+        gttRemap->unmap(engineInfo.pRingBuffer);
         engineInfo.pRingBuffer = nullptr;
     }
 }