Fix inconsistent fused op name

This commit is contained in:
Jianyu Wei 2021-05-20 03:04:50 -04:00
Родитель ef4e4e7582
Коммит 2a4f307f71
2 изменённых файлов: 1 добавлений и 2 удалений

Просмотреть файл

@ -49,7 +49,7 @@ class RuleReader:
'inbounds': [get_name(i - 1)] if i > 0 else [],
'outbounds': [get_name(i + 1)] if i < len(ops) - 1 else [],
}
self.fusion_units['_'.join(ops)] = Grapher(graph=fusion_unit)
self.fusion_units['-'.join(ops)] = Grapher(graph=fusion_unit)
def _parse_multiop_block(self):
for block in self.multiop_blocks:

Просмотреть файл

@ -43,7 +43,6 @@ class FusionAwareGraph:
else:
self._outbounds[node].update(self._outbounds[outnode])
def mark_ready(self, node):
self._ready[node] = True