misc: Proper type check and import for PortRef

Rewriting the type checking around PortRef, which was interacting strangely
with other Python scripts.

Tested-by: stephan.diestelhorst@arm.com

diff --git a/src/python/m5/util/dot_writer.py b/src/python/m5/util/dot_writer.py
index a6f721a0ca89a473f9ded709c4188f742f73a440..ab4687f7f25a5162e58209385147393f27ac778d 100644 (file)
--- 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: