Files
compute-runtime/shared/source/helpers/addressing_mode_helper.cpp
Jaroslaw Warchulski 45bbb7d6c4 fix: check all args in checkKernelContainsStatefulAccess function
Related-To: NEO-16281, NEO-16405
Signed-off-by: Jaroslaw Warchulski <jaroslaw.warchulski@intel.com>
2025-11-04 18:44:09 +01:00

89 lines
3.3 KiB
C++

/*
* Copyright (C) 2022-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "shared/source/helpers/addressing_mode_helper.h"
#include "shared/source/debug_settings/debug_settings_manager.h"
#include "shared/source/execution_environment/root_device_environment.h"
#include "shared/source/helpers/compiler_product_helper.h"
#include "shared/source/program/kernel_info.h"
namespace NEO::AddressingModeHelper {
bool failBuildProgramWithBufferStatefulAccess(const RootDeviceEnvironment &rootDeviceEnvironment) {
const auto &compilerProductHelper = rootDeviceEnvironment.getHelper<CompilerProductHelper>();
auto failBuildProgram = compilerProductHelper.failBuildProgramWithBufferStatefulAccessPreference();
if (NEO::debugManager.flags.FailBuildProgramWithStatefulAccess.get() != -1) {
failBuildProgram = static_cast<bool>(NEO::debugManager.flags.FailBuildProgramWithStatefulAccess.get());
}
auto forceToStatelessRequired = compilerProductHelper.isForceToStatelessRequired();
return failBuildProgram && forceToStatelessRequired;
}
inline bool argPointerIsStateful(const ArgDescriptor &arg) {
return arg.is<NEO::ArgDescriptor::argTPointer>() &&
(NEO::isValidOffset(arg.as<NEO::ArgDescPointer>().bindless) ||
NEO::isValidOffset(arg.as<NEO::ArgDescPointer>().bindful));
}
bool containsBufferStatefulAccess(const KernelDescriptor &kernelDescriptor, bool skipLastExplicitArg) {
auto size = static_cast<int32_t>(kernelDescriptor.payloadMappings.explicitArgs.size());
if (skipLastExplicitArg) {
size--;
}
for (auto i = 0; i < size; i++) {
if (argPointerIsStateful(kernelDescriptor.payloadMappings.explicitArgs[i])) {
return true;
}
}
return false;
}
bool containsBufferStatefulAccess(const std::vector<KernelInfo *> &kernelInfos, bool skipLastExplicitArg) {
for (const auto &kernelInfo : kernelInfos) {
if (containsBufferStatefulAccess(kernelInfo->kernelDescriptor, skipLastExplicitArg)) {
return true;
}
}
return false;
}
bool argIsStateful(const ArgDescriptor &arg) {
if (arg.is<NEO::ArgDescriptor::argTPointer>()) {
return (NEO::isValidOffset(arg.as<NEO::ArgDescPointer>().bindless) || NEO::isValidOffset(arg.as<NEO::ArgDescPointer>().bindful));
} else if (arg.is<NEO::ArgDescriptor::argTImage>()) {
return (NEO::isValidOffset(arg.as<NEO::ArgDescImage>().bindless) || NEO::isValidOffset(arg.as<NEO::ArgDescImage>().bindful));
} else if (arg.is<NEO::ArgDescriptor::argTSampler>()) {
return (NEO::isValidOffset(arg.as<NEO::ArgDescSampler>().bindless) || NEO::isValidOffset(arg.as<NEO::ArgDescSampler>().bindful));
}
return false;
}
bool containsStatefulAccess(const KernelDescriptor &kernelDescriptor) {
auto size = static_cast<int32_t>(kernelDescriptor.payloadMappings.explicitArgs.size());
for (auto i = 0; i < size; i++) {
if (argIsStateful(kernelDescriptor.payloadMappings.explicitArgs[i])) {
return true;
}
}
return false;
}
bool containsBindlessKernel(const std::vector<KernelInfo *> &kernelInfos) {
for (const auto &kernelInfo : kernelInfos) {
if (NEO::KernelDescriptor::isBindlessAddressingKernel(kernelInfo->kernelDescriptor)) {
return true;
}
}
return false;
}
} // namespace NEO::AddressingModeHelper