mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-22 10:17:01 +08:00
e357e7e404
taking into account the max work group limit when simd1 don't use the same limitation as for simd32 Resolves: NEO-14922 Related-To: NEO-11881 Signed-off-by: Katarzyna Cencelewska <katarzyna.cencelewska@intel.com>
37 lines
1.2 KiB
C++
37 lines
1.2 KiB
C++
/*
|
|
* Copyright (C) 2018-2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/execution_environment/root_device_environment.h"
|
|
#include "shared/source/helpers/gfx_core_helper.h"
|
|
#include "shared/source/helpers/local_id_gen.h"
|
|
#include "shared/source/helpers/simd_helper.h"
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
|
|
namespace NEO {
|
|
class LinearStream;
|
|
|
|
struct PerThreadDataHelper {
|
|
static inline size_t getPerThreadDataSizeTotal(
|
|
uint32_t simd,
|
|
uint32_t grfSize,
|
|
uint32_t grfCount,
|
|
uint32_t numChannels,
|
|
size_t localWorkSize,
|
|
const RootDeviceEnvironment &rootDeviceEnvironment) {
|
|
auto perThreadSizeLocalIDs = static_cast<size_t>(getPerThreadSizeLocalIDs(simd, grfSize, numChannels));
|
|
if (isSimd1(simd)) {
|
|
return perThreadSizeLocalIDs * localWorkSize;
|
|
}
|
|
auto &gfxCoreHelper = rootDeviceEnvironment.getHelper<NEO::GfxCoreHelper>();
|
|
return perThreadSizeLocalIDs * gfxCoreHelper.calculateNumThreadsPerThreadGroup(simd, static_cast<uint32_t>(localWorkSize), grfCount, rootDeviceEnvironment);
|
|
}
|
|
}; // namespace PerThreadDataHelper
|
|
} // namespace NEO
|