refactor: extend patchRegionParams with group count
Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
This commit is contained in:
parent
837d6f5ad1
commit
7aee5e1b7e
|
@ -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) {
|
||||
|
|
|
@ -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;
|
||||
|
|
|
@ -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;
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue