Expand adjustPipelineSelect parameters

Signed-off-by: Jaime Arteaga <[email protected]>

Author: Jaime Arteaga

shared/source/command_container/cmdcontainer.cpp

@@ -147,6 +147,7 @@ void CommandContainer::reset() {
     iddBlock = nullptr;
     nextIddInBlock = this->getNumIddPerBlock();
     lastSentNumGrfRequired = 0;
+    lastPipelineSelectModeRequired = false;
 }
 
 void *CommandContainer::getHeapSpaceAllowGrow(HeapType heapType,

shared/source/command_container/cmdcontainer.h

@@ -86,6 +86,7 @@ class CommandContainer : public NonCopyableOrMovableClass {
     uint32_t slmSize = std::numeric_limits<uint32_t>::max();
     uint32_t nextIddInBlock = 0;
     uint32_t lastSentNumGrfRequired = 0;
+    bool lastPipelineSelectModeRequired = false;
 
     Device *getDevice() const { return device; }
 

shared/source/command_container/command_encoder.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2020 Intel Corporation
+ * Copyright (C) 2020-2021 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -256,7 +256,7 @@ struct EncodeComputeMode {
     using STATE_COMPUTE_MODE = typename GfxFamily::STATE_COMPUTE_MODE;
     static void adjustComputeMode(LinearStream &csr, uint32_t numGrfRequired, void *const stateComputeModePtr, bool isMultiOsContextCapable);
 
-    static void adjustPipelineSelect(CommandContainer &container, uint32_t numGrfRequired);
+    static void adjustPipelineSelect(CommandContainer &container, const NEO::KernelDescriptor &kernelDescriptor);
 };
 
 template <typename GfxFamily>

shared/test/unit_test/fixtures/command_container_fixture.h

@@ -7,6 +7,7 @@
 
 #pragma once
 #include "shared/source/command_container/command_encoder.h"
+#include "shared/source/kernel/kernel_descriptor.h"
 #include "shared/test/unit_test/fixtures/device_fixture.h"
 
 #include "test.h"
@@ -31,6 +32,7 @@ class CommandEncodeStatesFixture : public DeviceFixture {
         DeviceFixture::TearDown();
     }
     std::unique_ptr<MyMockCommandContainer> cmdContainer;
+    KernelDescriptor descriptor;
 };
 
 } // namespace NEO