From fe75ead92e0f4a6a53dd961e3b154f7d19cc80e7 Mon Sep 17 00:00:00 2001 From: Michael Lazos Date: Wed, 30 Apr 2025 13:08:17 -0700 Subject: [PATCH] Import pydot lazily (#2248) --- python/cutlass/backend/evt/passes/graph_drawer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/cutlass/backend/evt/passes/graph_drawer.py b/python/cutlass/backend/evt/passes/graph_drawer.py index 4e1e094e..fd05bd92 100644 --- a/python/cutlass/backend/evt/passes/graph_drawer.py +++ b/python/cutlass/backend/evt/passes/graph_drawer.py @@ -29,11 +29,11 @@ # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. # ################################################################################################# +from __future__ import annotations import subprocess from cutlass_library import DataTypeTag -import pydot from cutlass.backend.evt.ir.dag_ir import DAGIR @@ -113,6 +113,7 @@ class EVTGraphDrawer: graph: DAGIR, name: str ): + import pydot dot_graph = pydot.Dot(name, randir="TB") for node in graph.nodes_meta: style = self._get_node_style(node)