Enhance the list of one time AUB writable types.

Change-Id: I1eba658aa83e0d3e44009e7eca928f5916d6f9ce
Signed-off-by: Mrozek, Michal <[email protected]>

Author: MichalMrozek

runtime/aub/aub_helper.h

@@ -17,6 +17,12 @@ class AubHelper : public NonCopyableOrMovableClass {
   public:
     static bool isOneTimeAubWritableAllocationType(const GraphicsAllocation::AllocationType &type) {
         switch (type) {
+        case GraphicsAllocation::AllocationType::PIPE:
+        case GraphicsAllocation::AllocationType::CONSTANT_SURFACE:
+        case GraphicsAllocation::AllocationType::GLOBAL_SURFACE:
+        case GraphicsAllocation::AllocationType::KERNEL_ISA:
+        case GraphicsAllocation::AllocationType::PRIVATE_SURFACE:
+        case GraphicsAllocation::AllocationType::SCRATCH_SURFACE:
         case GraphicsAllocation::AllocationType::BUFFER:
         case GraphicsAllocation::AllocationType::BUFFER_HOST_MEMORY:
         case GraphicsAllocation::AllocationType::BUFFER_COMPRESSED:

unit_tests/command_stream/aub_command_stream_receiver_1_tests.cpp

@@ -724,6 +724,12 @@ HWTEST_F(AubCommandStreamReceiverTests, givenAubCommandStreamReceiverWhenWriteMe
     auto gfxAllocation = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{MemoryConstants::pageSize});
 
     const GraphicsAllocation::AllocationType onlyOneTimeAubWritableTypes[] = {
+        GraphicsAllocation::AllocationType::PIPE,
+        GraphicsAllocation::AllocationType::CONSTANT_SURFACE,
+        GraphicsAllocation::AllocationType::GLOBAL_SURFACE,
+        GraphicsAllocation::AllocationType::KERNEL_ISA,
+        GraphicsAllocation::AllocationType::PRIVATE_SURFACE,
+        GraphicsAllocation::AllocationType::SCRATCH_SURFACE,
         GraphicsAllocation::AllocationType::BUFFER,
         GraphicsAllocation::AllocationType::BUFFER_HOST_MEMORY,
         GraphicsAllocation::AllocationType::BUFFER_COMPRESSED,