fix: move getting scratchPtr offset method to ImplicitArgs class

Related-To: NEO-16649
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>

Author: JablonskiMateusz

shared/source/command_container/command_encoder.h

@@ -309,7 +309,6 @@ struct EncodeDispatchKernel : public EncodeDispatchKernelBase<GfxFamily> {
     static void patchScratchAddressInImplicitArgs(ImplicitArgs &implicitArgs, uint64_t scratchAddress, bool scratchPtrPatchingRequired);
 
     static size_t getInlineDataOffset(EncodeDispatchKernelArgs &args);
-    static size_t getScratchPtrOffsetOfImplicitArgs();
 
     template <typename WalkerType>
     static void forceComputeWalkerPostSyncFlushWithWrite(WalkerType &walkerCmd);

shared/source/command_container/command_encoder_from_xe_hpg_core_to_xe3_core.inl

@@ -11,11 +11,6 @@
 
 namespace NEO {
 
-template <typename Family>
-size_t EncodeDispatchKernel<Family>::getScratchPtrOffsetOfImplicitArgs() {
-    return 0;
-}
-
 template <typename Family>
 void EncodeSurfaceState<Family>::setPitchForScratch(R_SURFACE_STATE *surfaceState, uint32_t pitch, const ProductHelper &productHelper) {
     surfaceState->setSurfacePitch(pitch);

shared/source/gen12lp/command_encoder_gen12lp.cpp

@@ -692,11 +692,6 @@ void EncodeDispatchKernel<Family>::encodeThreadGroupDispatch(InterfaceDescriptor
                                                              const uint32_t grfCount, const uint32_t threadsPerThreadGroup, WalkerType &walkerCmd) {
 }
 
-template <typename Family>
-size_t EncodeDispatchKernel<Family>::getScratchPtrOffsetOfImplicitArgs() {
-    return 0;
-}
-
 template <typename Family>
 void EncodeSurfaceState<Family>::setPitchForScratch(R_SURFACE_STATE *surfaceState, uint32_t pitch, const ProductHelper &productHelper) {
     surfaceState->setSurfacePitch(pitch);

shared/source/kernel/definitions/implicit_args.h

@@ -361,6 +361,13 @@ struct alignas(32) ImplicitArgs {
             z = v2.enqueuedLocalSizeZ;
         }
     }
+
+    size_t getScratchPtrOffset() const {
+        if (v1.header.structVersion == 1) {
+            return offsetof(ImplicitArgsV1, scratchPtr);
+        }
+        return 0;
+    }
 };
 
 } // namespace NEO

shared/test/unit_test/command_container/command_encoder_tests.cpp

@@ -1061,12 +1061,6 @@ HWTEST2_F(CommandEncoderTests, givenInterfaceDescriptorWhenEncodeEuSchedulingPol
     }
 }
 
-HWTEST_F(CommandEncoderTests, whenGetScratchPtrOffsetOfImplicitArgsIsCalledThenZeroIsReturned) {
-
-    auto scratchOffset = EncodeDispatchKernel<FamilyType>::getScratchPtrOffsetOfImplicitArgs();
-    EXPECT_EQ(0u, scratchOffset);
-}
-
 HWTEST_F(CommandEncoderTests, givenInOrderExecInfoWhenAggregatedEventUsageCounterIsUsedThenVerifyCorrectBehavior) {
     MockDevice mockDevice;
 

shared/test/unit_test/kernel/implicit_args_helper_tests.cpp

@@ -248,6 +248,8 @@ TEST(ImplicitArgsV0Test, givenImplicitArgsV0WhenSettingFieldsThenCorrectFieldsAr
     EXPECT_EQ(0x1000123400u, implicitArgs.v0.rtGlobalBufferPtr);
 
     EXPECT_EQ(32u, implicitArgs.getSimdWidth().value());
+
+    EXPECT_EQ(0u, implicitArgs.getScratchPtrOffset());
 }
 
 TEST(ImplicitArgsV1Test, givenImplicitArgsV1WhenSettingFieldsThenCorrectFieldsAreSet) {
@@ -299,6 +301,8 @@ TEST(ImplicitArgsV1Test, givenImplicitArgsV1WhenSettingFieldsThenCorrectFieldsAr
     EXPECT_EQ(0x1234000u, implicitArgs.v1.syncBufferPtr);
 
     EXPECT_EQ(32u, implicitArgs.getSimdWidth().value());
+
+    EXPECT_EQ(offsetof(ImplicitArgsV1, scratchPtr), implicitArgs.getScratchPtrOffset());
 }
 
 TEST(ImplicitArgsV2Test, givenImplicitArgsV2WhenSettingFieldsThenCorrectFieldsAreSet) {
@@ -348,6 +352,8 @@ TEST(ImplicitArgsV2Test, givenImplicitArgsV2WhenSettingFieldsThenCorrectFieldsAr
     EXPECT_EQ(16u, implicitArgs.v2.numWorkDim);
     EXPECT_EQ(0x1000123400u, implicitArgs.v2.rtGlobalBufferPtr);
     EXPECT_EQ(0x1234000u, implicitArgs.v2.syncBufferPtr);
+
+    EXPECT_EQ(0u, implicitArgs.getScratchPtrOffset());
 }
 
 TEST(ImplicitArgsV1Test, givenImplicitArgsWithUnknownVersionWhenSettingFieldsThenFieldsAreNotPopulated) {

shared/test/unit_test/mocks/mock_dispatch_kernel_encoder_interface.h

@@ -25,13 +25,14 @@ struct MockDispatchKernelEncoder : public DispatchKernelEncoderI {
         return numThreadsPerThreadGroup;
     }
 
-    NEO::ImplicitArgs *getImplicitArgs() const override { return nullptr; }
+    ImplicitArgs *getImplicitArgs() const override { return implicitArgsPtr; }
 
     void patchBindlessOffsetsInCrossThreadData(uint64_t bindlessSurfaceStateBaseOffset) const override { return; };
     void patchSamplerBindlessOffsetsInCrossThreadData(uint64_t samplerStateOffset) const override {
         samplerStateOffsetPassed = samplerStateOffset;
     }
 
+    ImplicitArgs *implicitArgsPtr = nullptr;
     MockGraphicsAllocation mockAllocation{};
     static constexpr uint32_t crossThreadSize = 0x40;
     static constexpr uint32_t perThreadSize = 0x20;