1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
/*
* 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
|
