check source type

Author: cse0001

visualdl/component/graph/graph_component.py

@@ -464,7 +464,8 @@ def get_sub_ops(op, op_name, all_ops, all_vars):
             all_ops[sub_op_name]['is_leaf_node'] = True
             now_var = utils.gen_var_name(sub_op.results())
             for source in sub_op.operands_source():
-                if str(source.type()) == '<<NULL TYPE>>':
+                if not source.type():
+                # if source.type() == Value().type():
                     continue
                 input_name = utils.gen_var_name(source)
                 if input_name not in all_vars.keys():
@@ -655,7 +656,8 @@ def analyse_pir(program):
             all_ops[op_name]['is_leaf_node'] = True
             now_var = utils.gen_var_name(op.results())
             for source in op.operands_source():
-                if str(source.type()) == '<<NULL TYPE>>':
+                if not source.type():
+                # if source.type() == Value().type():
                     continue
                 input_name = utils.gen_var_name(source)
                 if input_name not in all_vars.keys():