fix: use gpuAddress when patching sync buffer

Related-To: NEO-11533
Related-To: HSD-18039788811
Signed-off-by: Fabian Zwoliński <[email protected]>
Source: 6a50320

Author: fzwolinski

level_zero/core/source/kernel/kernel_imp.cpp

@@ -1182,7 +1182,7 @@ void KernelImp::patchSyncBuffer(NEO::GraphicsAllocation *gfxAllocation, size_t b
     this->internalResidencyContainer.push_back(gfxAllocation);
     NEO::patchPointer(ArrayRef<uint8_t>(crossThreadData.get(), crossThreadDataSize),
                       this->getImmutableData()->getDescriptor().payloadMappings.implicitArgs.syncBufferAddress,
-                      static_cast<uintptr_t>(ptrOffset(gfxAllocation->getGpuAddressToPatch(), bufferOffset)));
+                      static_cast<uintptr_t>(ptrOffset(gfxAllocation->getGpuAddress(), bufferOffset)));
 }
 
 void KernelImp::patchRegionGroupBarrier(NEO::GraphicsAllocation *gfxAllocation, size_t bufferOffset) {

level_zero/core/test/unit_tests/sources/kernel/test_kernel.cpp

@@ -4169,5 +4169,37 @@ TEST_F(BindlessKernelTest, givenBindlessKernelWithInlineSamplersWhenPatchingSamp
     EXPECT_EQ(patchValue2, crossThreadData[6]);
 }
 
+using KernelSyncBufferTest = Test<ModuleFixture>;
+
+TEST_F(KernelSyncBufferTest, GivenSyncBufferArgWhenPatchingSyncBufferThenPtrIsCorrectlyPatchedInCrossThreadData) {
+    Mock<KernelImp> kernel;
+    neoDevice->incRefInternal();
+
+    Mock<Module> mockModule(device, nullptr);
+    kernel.module = &mockModule;
+
+    kernel.crossThreadData = std::make_unique<uint8_t[]>(64);
+    kernel.crossThreadDataSize = 64;
+
+    auto &syncBuffer = kernel.immutableData.kernelDescriptor->payloadMappings.implicitArgs.syncBufferAddress;
+    syncBuffer.stateless = 0x8;
+    syncBuffer.pointerSize = 8;
+
+    NEO::MockGraphicsAllocation alloc;
+    alloc.setGpuPtr(0xffff800300060000);
+    alloc.setGpuBaseAddress(0xffff800300000000);
+    alloc.allocationOffset = 0x0;
+
+    size_t bufferOffset = 0u;
+
+    kernel.patchSyncBuffer(&alloc, bufferOffset);
+
+    auto patchValue = *reinterpret_cast<uint64_t *>(ptrOffset(kernel.crossThreadData.get(), syncBuffer.stateless));
+    auto expectedPatchValue = ptrOffset(alloc.getGpuAddress(), bufferOffset);
+    EXPECT_EQ(expectedPatchValue, patchValue);
+
+    neoDevice->decRefInternal();
+}
+
 } // namespace ult
 } // namespace L0