Added scratch size check

Related-To: NEO-7508
Signed-off-by: Andrzej Koska <andrzej.koska@intel.com>
This commit is contained in:
Andrzej Koska
2022-11-17 14:27:52 +00:00
committed by Compute-Runtime-Automation
parent fd691cd33c
commit 90034d4173
14 changed files with 183 additions and 7 deletions

View File

@@ -803,6 +803,9 @@ ze_result_t KernelImp::initialize(const ze_kernel_desc_t *desc) {
const auto &hwInfoConfig = *NEO::HwInfoConfig::get(hwInfo.platform.eProductFamily);
auto &kernelDescriptor = kernelImmData->getDescriptor();
auto ret = NEO::KernelHelper::checkIfThereIsSpaceForScratchOrPrivate(kernelDescriptor.kernelAttributes, neoDevice);
if (ret == NEO::KernelHelper::ErrorCode::INVALID_KERNEL) {
return ZE_RESULT_ERROR_INVALID_NATIVE_BINARY;
}
if (ret == NEO::KernelHelper::ErrorCode::OUT_OF_DEVICE_MEMORY) {
return ZE_RESULT_ERROR_OUT_OF_DEVICE_MEMORY;
}