compute-runtime/opencl/test/unit_test/mocks/mock_printf_handler.h

32 lines
779 B
C++

/*
* Copyright (C) 2022 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "opencl/source/program/printf_handler.h"
using namespace NEO;
class MockPrintfHandler : public PrintfHandler {
public:
using PrintfHandler::PrintfHandler;
using PrintfHandler::printfSurface;
using PrintfHandler::printfSurfaceInitialDataSizePtr;
MockPrintfHandler(Device &device) : PrintfHandler{device} {}
~MockPrintfHandler() override = default;
bool printEnqueueOutput() override {
if (callBasePrintEnqueueOutput) {
return PrintfHandler::printEnqueueOutput();
}
return printEnqueueOutputReturnValue;
}
bool callBasePrintEnqueueOutput = false;
bool printEnqueueOutputReturnValue = false;
};