diff --git a/level_zero/core/source/cmdlist/cmdlist_hw_xehp_and_later.inl b/level_zero/core/source/cmdlist/cmdlist_hw_xehp_and_later.inl index adc73c9dc7..b58112695f 100644 --- a/level_zero/core/source/cmdlist/cmdlist_hw_xehp_and_later.inl +++ b/level_zero/core/source/cmdlist/cmdlist_hw_xehp_and_later.inl @@ -164,7 +164,7 @@ ze_result_t CommandListCoreFamily::appendLaunchKernelWithParams(K auto kernelPreemptionMode = obtainKernelPreemptionMode(kernel); kernel->patchGlobalOffset(); - kernel->patchRegionParams(launchParams); + kernel->patchRegionParams(launchParams, threadGroupDimensions); this->allocateOrReuseKernelPrivateMemoryIfNeeded(kernel, kernelDescriptor.kernelAttributes.perHwThreadPrivateMemorySize); if (launchParams.isIndirect) { diff --git a/level_zero/core/source/kernel/kernel.h b/level_zero/core/source/kernel/kernel.h index a34125eef6..a644306e95 100644 --- a/level_zero/core/source/kernel/kernel.h +++ b/level_zero/core/source/kernel/kernel.h @@ -143,7 +143,7 @@ struct Kernel : _ze_kernel_handle_t, virtual NEO::DispatchKernelEncoderI, NEO::N virtual uint32_t *getGlobalOffsets() = 0; virtual ze_result_t setGlobalOffsetExp(uint32_t offsetX, uint32_t offsetY, uint32_t offsetZ) = 0; virtual void patchGlobalOffset() = 0; - virtual void patchRegionParams(const CmdListKernelLaunchParams &launchParams) = 0; + virtual void patchRegionParams(const CmdListKernelLaunchParams &launchParams, const ze_group_count_t &threadGroupDimensions) = 0; virtual uint32_t suggestMaxCooperativeGroupCount(NEO::EngineGroupType engineGroupType, bool forceSingleTileQuery) = 0; virtual ze_result_t setCacheConfig(ze_cache_config_flags_t flags) = 0; diff --git a/level_zero/core/source/kernel/kernel_imp.h b/level_zero/core/source/kernel/kernel_imp.h index 4116bdd1db..cd5b499528 100644 --- a/level_zero/core/source/kernel/kernel_imp.h +++ b/level_zero/core/source/kernel/kernel_imp.h @@ -59,7 +59,7 @@ struct KernelImp : Kernel { ze_result_t setArgumentValue(uint32_t argIndex, size_t argSize, const void *pArgValue) override; void setGroupCount(uint32_t groupCountX, uint32_t groupCountY, uint32_t groupCountZ) override; - void patchRegionParams(const CmdListKernelLaunchParams &launchParams) override; + void patchRegionParams(const CmdListKernelLaunchParams &launchParams, const ze_group_count_t &threadGroupDimensions) override; ze_result_t setGroupSize(uint32_t groupSizeX, uint32_t groupSizeY, uint32_t groupSizeZ) override; diff --git a/level_zero/core/source/kernel/kernel_imp_helper.cpp b/level_zero/core/source/kernel/kernel_imp_helper.cpp index 9316527ee2..e317cf27e9 100644 --- a/level_zero/core/source/kernel/kernel_imp_helper.cpp +++ b/level_zero/core/source/kernel/kernel_imp_helper.cpp @@ -1,5 +1,5 @@ /* - * Copyright (C) 2022-2024 Intel Corporation + * Copyright (C) 2022-2025 Intel Corporation * * SPDX-License-Identifier: MIT * @@ -12,6 +12,6 @@ namespace L0 { KernelExt *KernelImp::getExtension(uint32_t extensionType) { return nullptr; } -void KernelImp::patchRegionParams(const CmdListKernelLaunchParams &launchParams) {} +void KernelImp::patchRegionParams(const CmdListKernelLaunchParams &launchParams, const ze_group_count_t &threadGroupDimensions) {} } // namespace L0