diff --git a/opencl/source/helpers/hardware_commands_helper_base.inl b/opencl/source/helpers/hardware_commands_helper_base.inl index 50a0c4fc86..3c3a9ede32 100644 --- a/opencl/source/helpers/hardware_commands_helper_base.inl +++ b/opencl/source/helpers/hardware_commands_helper_base.inl @@ -76,7 +76,7 @@ size_t HardwareCommandsHelper::getSizeRequiredIOH(const Kernel &kerne if (pImplicitArgs) { size += ImplicitArgsHelper::getSizeForImplicitArgsPatching(pImplicitArgs, kernelDescriptor, isHwLocalIdGeneration, rootDeviceEnvironment); } - return alignUp(size, GfxFamily::indirectDataAlignment); + return alignUp(size, rootDeviceEnvironment.getHelper().getIOHAlignment()); } template diff --git a/opencl/test/unit_test/command_queue/dispatch_walker_tests.cpp b/opencl/test/unit_test/command_queue/dispatch_walker_tests.cpp index 7bbc899c45..ec4e7f2f30 100644 --- a/opencl/test/unit_test/command_queue/dispatch_walker_tests.cpp +++ b/opencl/test/unit_test/command_queue/dispatch_walker_tests.cpp @@ -1385,7 +1385,7 @@ HWTEST_F(DispatchWalkerTest, WhenKernelRequiresImplicitArgsThenIohRequiresMoreSp HardwareCommandsHelper::getPerThreadDataSizeTotal(simdSize, grfSize, numGrf, numChannels, Math::computeTotalElementsCount(workGroupSize), false, rootDeviceEnvironment) + ImplicitArgsHelper::getSizeForImplicitArgsPatching(kernelWithImplicitArgs.getImplicitArgs(), kernelWithImplicitArgs.getDescriptor(), false, rootDeviceEnvironment); - size = alignUp(size, MemoryConstants::cacheLineSize); + size = alignUp(size, pClDevice->getGfxCoreHelper().getIOHAlignment()); EXPECT_EQ(size, iohSizeWithImplicitArgs); } }