leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

v4.3 tag release update. (#2789)

Browse Source
This commit is contained in:
Junkai-Wu
2025-11-21 09:49:44 +08:00
committed by GitHub
parent 406e078b29
commit 8cd5bef43a
225 changed files with 23229 additions and 2813 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
python/cutlass_cppgen/backend/evt/passes/graph_drawer.py
View File

@@ -99,11 +99,10 @@ class EVTGraphDrawer:
stride = node.tensor.stride
label += f"|shape={shape}|stride={stride}"
if hasattr(node, "store_tensor"):
if node.store_tensor is not None:
store_shape = node.store_tensor.shape
store_stride = node.store_tensor.stride
label += f"|store_shape={store_shape}|stride_stride={store_stride}"
if hasattr(node, "store_tensor") and node.store_tensor is not None:
store_shape = node.store_tensor.shape
store_stride = node.store_tensor.stride
label += f"|store_shape={store_shape}|store_stride={store_stride}"
label += "}"
return label
@@ -114,7 +113,7 @@ class EVTGraphDrawer:
name: str
):
import pydot
dot_graph = pydot.Dot(name, randir="TB")
dot_graph = pydot.Dot(name, rankdir="TB")
for node in graph.nodes_meta:
style = self._get_node_style(node)
label = self._get_node_label(node)
@@ -133,11 +132,11 @@ class EVTGraphDrawer:
return dot_graph
def get_dot_graph(self) -> pydot.Dot:
def get_dot_graph(self) -> "pydot.Dot":
return [(key, self.get_dot_graph_by_name(key)) for key in self._dot_graphs.keys()]
def get_dot_graph_by_name(self, name) -> pydot.Dot:
def get_dot_graph_by_name(self, name) -> "pydot.Dot":
return self._dot_graphs[name]
def get_main_dot_graph(self) -> pydot.Dot:
def get_main_dot_graph(self) -> "pydot.Dot":
return self._dot_graphs[self._name]