summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/python/m5/util/dot_writer.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/python/m5/util/dot_writer.py b/src/python/m5/util/dot_writer.py
index a6f721a0c..ab4687f7f 100644
--- a/src/python/m5/util/dot_writer.py
+++ b/src/python/m5/util/dot_writer.py
@@ -58,6 +58,7 @@
import m5, os, re
from m5.SimObject import isRoot, isSimObjectVector
+from m5.params import PortRef
from m5.util import warn
try:
import pydot
@@ -106,7 +107,7 @@ def dot_create_edges(simNode, callgraph):
full_port_name = full_path + "_" + port_name
port_node = dot_create_node(simNode, full_port_name, port_name)
# create edges
- if type(port) is m5.params.PortRef:
+ if isinstance(port, PortRef):
dot_add_edge(simNode, callgraph, full_port_name, port)
else:
for p in port.elements: