We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0f744be commit 0987146Copy full SHA for 0987146
py/torch_tensorrt/dynamo/conversion/_ConverterRegistry.py
@@ -481,7 +481,7 @@ def __getitem__(
481
return (
482
converters,
483
calling_convention,
484
- candidate.requires_output_allocator,
+ False,
485
)
486
487
raise KeyError(
py/torch_tensorrt/dynamo/lowering/passes/remove_num_users_is_0_nodes.py
@@ -16,7 +16,11 @@ def remove_num_users_is_0_nodes(
16
output_node = list(gm.graph.nodes)[-1]
17
18
for node in gm.graph.nodes:
19
- if node != output_node and len(node.users) == 0:
+ if (
20
+ node != output_node
21
+ and len(node.users) == 0
22
+ and len(node.all_input_nodes) > 0
23
+ ):
24
node_input = node.all_input_nodes[0]
25
node.replace_all_uses_with(node_input)
26
gm.graph.erase_node(node)
0 commit comments