Add test for getMaxParameterSizeFromIGC

Test is using MockCompilerInterface

Related-To: NEO-4851

Signed-off-by: Dominik Dabek <[email protected]>

Author: ddabek-i

shared/source/compiler_interface/compiler_interface.h

@@ -136,7 +136,7 @@ class CompilerInterface {
     MOCKABLE_VIRTUAL TranslationOutput::ErrorCode getSipKernelBinary(NEO::Device &device, SipKernelType type, std::vector<char> &retBinary,
                                                                      std::vector<char> &stateSaveAreaHeader);
 
-    CIF::RAII::UPtr_t<IGC::IgcFeaturesAndWorkaroundsTagOCL> getIgcFeaturesAndWorkarounds(const NEO::Device &device);
+    MOCKABLE_VIRTUAL CIF::RAII::UPtr_t<IGC::IgcFeaturesAndWorkaroundsTagOCL> getIgcFeaturesAndWorkarounds(const NEO::Device &device);
 
   protected:
     MOCKABLE_VIRTUAL bool initialize(std::unique_ptr<CompilerCache> &&cache, bool requireFcl);

shared/test/common/mocks/mock_compiler_interface.h

@@ -133,11 +133,20 @@ class MockCompilerInterface : public CompilerInterface {
         }
     }
 
+    CIF::RAII::UPtr_t<IGC::IgcFeaturesAndWorkaroundsTagOCL> getIgcFeaturesAndWorkarounds(const NEO::Device &device) override {
+        if (nullptr != igcFeaturesAndWorkaroundsTagOCL) {
+            return CIF::RAII::RetainAndPack<IGC::IgcFeaturesAndWorkaroundsTagOCL>(igcFeaturesAndWorkaroundsTagOCL);
+        } else {
+            return CompilerInterface::getIgcFeaturesAndWorkarounds(device);
+        }
+    };
+
     static std::vector<char> getDummyGenBinary();
     static void releaseDummyGenBinary();
 
     void (*lockListener)(MockCompilerInterface &compInt) = nullptr;
     void *lockListenerData = nullptr;
+    IGC::IgcFeaturesAndWorkaroundsTagOCL *igcFeaturesAndWorkaroundsTagOCL = nullptr;
     bool failCreateFclTranslationCtx = false;
     bool failCreateIgcTranslationCtx = false;
     bool failLoadFcl = false;

shared/test/common/mocks/mock_compilers.h

@@ -120,6 +120,11 @@ struct MockGTSystemInfo : MockCIF<IGC::GTSystemInfoTagOCL> {
 };
 
 struct MockIgcFeaturesAndWorkarounds : MockCIF<IGC::IgcFeaturesAndWorkaroundsTagOCL> {
+    uint32_t GetMaxOCLParamSize() const override {
+        return this->maxOCLParamSize;
+    };
+
+    uint32_t maxOCLParamSize = 0;
 };
 
 struct MockIgcOclTranslationCtx : MockCIF<IGC::IgcOclTranslationCtxTagOCL> {

shared/test/common/mocks/mock_device.h

@@ -96,6 +96,9 @@ class MockDevice : public RootDevice {
     }
 
     size_t getMaxParameterSizeFromIGC() const override {
+        if (callBaseGetMaxParameterSizeFromIGC) {
+            return Device::getMaxParameterSizeFromIGC();
+        }
         return maxParameterSizeFromIGC;
     }
 
@@ -180,6 +183,7 @@ class MockDevice : public RootDevice {
     bool isDebuggerActiveParentCall = true;
     bool isDebuggerActiveReturn = false;
     bool rtDispatchGlobalsForceAllocation = false;
+    bool callBaseGetMaxParameterSizeFromIGC = false;
     size_t maxParameterSizeFromIGC = 0u;
 };
 

shared/test/unit_test/device/neo_device_tests.cpp

@@ -9,6 +9,8 @@
 #include "shared/test/common/fixtures/device_fixture.h"
 #include "shared/test/common/helpers/default_hw_info.h"
 #include "shared/test/common/helpers/variable_backup.h"
+#include "shared/test/common/mocks/mock_compiler_interface.h"
+#include "shared/test/common/mocks/mock_compilers.h"
 #include "shared/test/common/mocks/mock_device.h"
 #include "shared/test/common/mocks/ult_device_factory.h"
 #include "shared/test/common/test_macros/test.h"
@@ -81,8 +83,19 @@ TEST_F(DeviceTest, whenGetRTDispatchGlobalsIsCalledWithZeroSizeAndMockAllocatorT
 
 using DeviceGetCapsTest = Test<DeviceFixture>;
 
-TEST_F(DeviceGetCapsTest, givenNonZeroMaxParameterSizeFromIGCwhenDeviceIsCreatedThenMaxParameterSizeIsSetCorrectly) {
-    pDevice->maxParameterSizeFromIGC = 1u;
+TEST_F(DeviceGetCapsTest, givenMockCompilerInterfaceWhenInitializeCapsIsCalledThenMaxParameterSizeIsSetCorrectly) {
+    auto pCompilerInterface = new MockCompilerInterface;
+    pDevice->getExecutionEnvironment()->rootDeviceEnvironments[pDevice->getRootDeviceIndex()]->compilerInterface.reset(pCompilerInterface);
+    pDevice->maxParameterSizeFromIGC = 2u;
+    pDevice->callBaseGetMaxParameterSizeFromIGC = true;
+    MockIgcFeaturesAndWorkarounds mockIgcFtrWa;
+    pCompilerInterface->igcFeaturesAndWorkaroundsTagOCL = &mockIgcFtrWa;
+
+    mockIgcFtrWa.maxOCLParamSize = 0u;
     pDevice->initializeCaps();
-    EXPECT_EQ(pDevice->getDeviceInfo().maxParameterSize, 1u);
-}
+    EXPECT_EQ(2048u, pDevice->getDeviceInfo().maxParameterSize);
+
+    mockIgcFtrWa.maxOCLParamSize = 1u;
+    pDevice->initializeCaps();
+    EXPECT_EQ(1u, pDevice->getDeviceInfo().maxParameterSize);
+}