MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / __init__

Method __init__

imperative/python/megengine/xla/ir_utils.py:117–144  ·  view source on GitHub ↗
(self, traced, func_name=None)

Source from the content-addressed store, hash-verified

115
116class TraceResult:
117 def __init__(self, traced, func_name=None) -> None:
118 self.func_name = func_name if func_name is not None else _default_func_name()
119 self.traced = traced
120 self.eqns = []
121 self.vars = {}
122 self.inputs = []
123 self.outputs = []
124 self.consts = []
125 self.custom_vid = 0
126
127 self.effects = []
128 self.has_rng_opr = False
129
130 for var in self.traced.vars:
131 self.add_var(var)
132 self.custom_vid = max(var.id + 1, self.custom_vid)
133
134 if var.kind == "external" and var.inp_mark:
135 self.inputs.append(var.id)
136
137 if var.data_required:
138 self.outputs.append(var.id)
139
140 if var.kind == "const":
141 self.consts.append(var.id)
142
143 for op in self.traced.ops:
144 self.eqns.append(op)
145
146 @property
147 def _var_inputs(self):

Callers

nothing calls this directly

Calls 4

add_varMethod · 0.95
_default_func_nameFunction · 0.85
maxFunction · 0.85
appendMethod · 0.45

Tested by

no test coverage detected