mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-01 12:33:12 +08:00
42364dd070
Implement handling for: - zeCommandListAppendLaunchKernelWithArguments - zeCommandListAppendLaunchKernelWithParameters Related-To: NEO-15377 Signed-off-by: Naklicki, Mateusz <mateusz.naklicki@intel.com>
70 lines
2.6 KiB
C++
70 lines
2.6 KiB
C++
/*
|
|
* Copyright (C) 2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "level_zero/ze_api.h"
|
|
|
|
#include "graph.h"
|
|
|
|
#include <optional>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
namespace L0 {
|
|
|
|
struct CommandList;
|
|
|
|
enum class GraphExportStyle : std::uint8_t {
|
|
detailed,
|
|
simple
|
|
};
|
|
|
|
class GraphDotExporter {
|
|
public:
|
|
explicit GraphDotExporter(GraphExportStyle style) : exportStyle(style) {}
|
|
|
|
ze_result_t exportToFile(const Graph &graph, const char *filePath) const;
|
|
|
|
protected:
|
|
std::string exportToString(const Graph &graph) const;
|
|
|
|
void writeHeader(std::ostringstream &dot) const;
|
|
void writeNodes(std::ostringstream &dot, const Graph &graph, uint32_t level, uint32_t subgraphId) const;
|
|
void writeSubgraphs(std::ostringstream &dot, const Graph &graph, uint32_t level) const;
|
|
void writeEdges(std::ostringstream &dot, const Graph &graph, uint32_t level, uint32_t subgraphId) const;
|
|
void writeSequentialEdges(std::ostringstream &dot, const Graph &graph, uint32_t level, uint32_t subgraphId) const;
|
|
void writeForkJoinEdges(std::ostringstream &dot, const Graph &graph, uint32_t level, uint32_t subgraphId) const;
|
|
void writeUnjoinedForkEdges(std::ostringstream &dot, const Graph &graph, uint32_t level, uint32_t subgraphId) const;
|
|
|
|
std::optional<uint32_t> findSubgraphIndex(std::span<Graph *const> subGraphs, const Graph *targetGraph) const;
|
|
std::optional<uint32_t> findSubgraphIndexByCommandList(std::span<Graph *const> subGraphs, const L0::CommandList *cmdList) const;
|
|
|
|
std::string getCommandNodeLabel(const Graph &graph, CapturedCommandId cmdId, const std::string_view indent) const;
|
|
std::string getCommandNodeAttributes(const Graph &graph, CapturedCommandId cmdId) const;
|
|
std::string generateNodeId(uint32_t level, uint32_t subgraphId, CapturedCommandId cmdId) const;
|
|
std::string generateSubgraphId(uint32_t level, uint32_t subgraphId) const;
|
|
std::string getSubgraphFillColor(uint32_t level) const;
|
|
|
|
private:
|
|
GraphExportStyle exportStyle;
|
|
};
|
|
|
|
namespace GraphDumpHelper {
|
|
|
|
std::string formatPointer(const void *ptr);
|
|
std::string formatGroupCount(const ze_group_count_t &groupCount);
|
|
std::string formatGroupSize(const ze_group_size_t &groupSize);
|
|
std::string_view getKernelName(ze_kernel_handle_t kernelHandle);
|
|
std::string_view getKernelName(const CapturedCommand &cmd);
|
|
void addKernelInformation(std::vector<std::pair<std::string, std::string>> ¶ms, ze_kernel_handle_t kernelHandle);
|
|
void addLaunchKernelExtensionParameters(std::vector<std::pair<std::string, std::string>> ¶ms, const void *pNext);
|
|
|
|
} // namespace GraphDumpHelper
|
|
|
|
} // namespace L0
|