Initial commit

Change-Id: I4bf1707bd3dfeadf2c17b0a7daff372b1925ebbd
2025-12-25 21:42:53 +08:00 · 2017-12-21 00:45:38 +01:00
commit 7e9ad41290
1350 changed files with 233156 additions and 0 deletions
--- a/runtime/program/printf_handler.cpp
+++ b/runtime/program/printf_handler.cpp
@@ -0,0 +1,81 @@
+/*
+ * Copyright (c) 2017, Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included
+ * in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include "printf_handler.h"
+
+#include "runtime/mem_obj/buffer.h"
+#include "runtime/program/print_formatter.h"
+#include "runtime/kernel/kernel.h"
+#include "runtime/helpers/dispatch_info.h"
+#include "runtime/helpers/ptr_math.h"
+#include "runtime/helpers/aligned_memory.h"
+#include "runtime/memory_manager/memory_manager.h"
+
+namespace OCLRT {
+
+PrintfHandler::PrintfHandler(Device &deviceArg) : device(deviceArg) {}
+
+PrintfHandler::~PrintfHandler() {
+    device.getMemoryManager()->freeGraphicsMemory(printfSurface);
+}
+
+PrintfHandler *PrintfHandler::create(const MultiDispatchInfo &multiDispatchInfo, Device &device) {
+    if (multiDispatchInfo.usesStatelessPrintfSurface() ||
+        (multiDispatchInfo.begin()->getKernel()->checkIfIsParentKernelAndBlocksUsesPrintf())) {
+        return new PrintfHandler(device);
+    }
+    return nullptr;
+}
+
+void PrintfHandler::prepareDispatch(const MultiDispatchInfo &multiDispatchInfo) {
+    auto printfSurfaceSize = device.getDeviceInfo().printfBufferSize;
+    if (printfSurfaceSize == 0) {
+        return;
+    }
+    kernel = multiDispatchInfo.begin()->getKernel();
+
+    printfSurface = device.getMemoryManager()->createGraphicsAllocationWithRequiredBitness(printfSurfaceSize, nullptr);
+
+    *reinterpret_cast<uint32_t *>(printfSurface->getUnderlyingBuffer()) = printfSurfaceInitialDataSize;
+
+    auto printfPatchAddress = ptrOffset(reinterpret_cast<uintptr_t *>(kernel->getCrossThreadData()),
+                                        kernel->getKernelInfo().patchInfo.pAllocateStatelessPrintfSurface->DataParamOffset);
+
+    patchWithRequiredSize(printfPatchAddress, kernel->getKernelInfo().patchInfo.pAllocateStatelessPrintfSurface->DataParamSize, (uintptr_t)printfSurface->getGpuAddressToPatch());
+    if (kernel->requiresSshForBuffers()) {
+        auto surfaceState = ptrOffset(reinterpret_cast<uintptr_t *>(kernel->getSurfaceStateHeap()),
+                                      kernel->getKernelInfo().patchInfo.pAllocateStatelessPrintfSurface->SurfaceStateHeapOffset);
+        void *addressToPatch = printfSurface->getUnderlyingBuffer();
+        size_t sizeToPatch = printfSurface->getUnderlyingBufferSize();
+        Buffer::setSurfaceState(&kernel->getContext(), surfaceState, sizeToPatch, addressToPatch, printfSurface);
+    }
+}
+
+void PrintfHandler::makeResident(CommandStreamReceiver &commandStreamReceiver) {
+    commandStreamReceiver.makeResident(*printfSurface);
+}
+
+void PrintfHandler::printEnqueueOutput() {
+    PrintFormatter printFormatter(*kernel, *printfSurface);
+    printFormatter.printKernelOutput();
+}
+} // namespace OCLRT