refactor: extend patchRegionParams with group count

Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
This commit is contained in:
Mateusz Hoppe 2025-02-25 17:31:49 +00:00 committed by Compute-Runtime-Automation
parent 837d6f5ad1
commit 7aee5e1b7e
4 changed files with 5 additions and 5 deletions

View File

@ -164,7 +164,7 @@ ze_result_t CommandListCoreFamily<gfxCoreFamily>::appendLaunchKernelWithParams(K
auto kernelPreemptionMode = obtainKernelPreemptionMode(kernel);
kernel->patchGlobalOffset();
kernel->patchRegionParams(launchParams);
kernel->patchRegionParams(launchParams, threadGroupDimensions);
this->allocateOrReuseKernelPrivateMemoryIfNeeded(kernel, kernelDescriptor.kernelAttributes.perHwThreadPrivateMemorySize);
if (launchParams.isIndirect) {

View File

@ -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;

View File

@ -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;

View File

@ -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