mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-23 03:01:20 +08:00
c44b600ec7
- assert buffer has header that needs to be accounted for while parsing the buffer and reading format strings. currentOffset in Printf Formatter must not exceed size in header. The offset is relative to the beginning of buffer, not the "begin" field in header Resolves: NEO-8237 Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
59 lines
1.9 KiB
C++
59 lines
1.9 KiB
C++
/*
|
|
* Copyright (C) 2023 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/assert_handler/assert_handler.h"
|
|
|
|
#include "shared/source/device/device.h"
|
|
#include "shared/source/helpers/abort.h"
|
|
#include "shared/source/memory_manager/allocation_properties.h"
|
|
#include "shared/source/memory_manager/memory_manager.h"
|
|
#include "shared/source/program/print_formatter.h"
|
|
|
|
namespace NEO {
|
|
AssertHandler::AssertHandler(Device *device) : device(device) {
|
|
auto rootDeviceIndex = device->getRootDeviceIndex();
|
|
assertBuffer = device->getMemoryManager()->allocateGraphicsMemoryWithProperties({rootDeviceIndex, assertBufferSize, AllocationType::ASSERT_BUFFER, device->getDeviceBitfield()});
|
|
|
|
AssertBufferHeader initialHeader = {};
|
|
initialHeader.size = assertBufferSize;
|
|
initialHeader.flags = 0;
|
|
initialHeader.begin = sizeof(AssertBufferHeader);
|
|
*reinterpret_cast<AssertBufferHeader *>(assertBuffer->getUnderlyingBuffer()) = initialHeader;
|
|
}
|
|
|
|
AssertHandler::~AssertHandler() {
|
|
device->getMemoryManager()->freeGraphicsMemory(assertBuffer);
|
|
}
|
|
|
|
bool AssertHandler::checkAssert() const {
|
|
return reinterpret_cast<AssertBufferHeader *>(assertBuffer->getUnderlyingBuffer())->flags != 0;
|
|
}
|
|
|
|
void AssertHandler::printMessage() const {
|
|
|
|
auto messageBuffer = static_cast<uint8_t *>(assertBuffer->getUnderlyingBuffer());
|
|
auto messageBufferSize = static_cast<uint32_t>(assertBuffer->getUnderlyingBufferSize());
|
|
|
|
NEO::PrintFormatter printfFormatter{
|
|
messageBuffer,
|
|
messageBufferSize,
|
|
false,
|
|
nullptr};
|
|
printfFormatter.setInitialOffset(offsetof(AssertBufferHeader, begin));
|
|
|
|
printfFormatter.printKernelOutput([](char *str) { printToStderr(str); });
|
|
}
|
|
|
|
void AssertHandler::printAssertAndAbort() {
|
|
std::lock_guard<std::mutex> lock(this->mtx);
|
|
if (checkAssert()) {
|
|
printMessage();
|
|
abortExecution();
|
|
}
|
|
}
|
|
} // namespace NEO
|