microsoft / nn-Meter

A DNN inference latency prediction toolkit for accurately modeling and predicting the latency on diverse edge devices.

Why removed nodes are not removed from inbound

YuqiData-UW opened this issue a year ago · comments

YuqiData-UW commented a year ago

If not they are not removed,

nn-Meter/nn_meter/kernel_detector/utils/fusion_aware_graph.py

Line 21 in 759854e

{reverse[outbound] for outbound in self._model_graph.get_node_outbounds(name)}

can access non-existing node from graph.

I made the following change in my own fork.
YuqiData-UW@167edcd#diff-dd5660ddb30cf4ee6c6bc08175136958f8683dab0f9466a2aa744ce05558ac46R74