Skip to content

Commit 0987146

Browse files
committed
fix bugs from CI
1 parent 0f744be commit 0987146

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

py/torch_tensorrt/dynamo/conversion/_ConverterRegistry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -481,7 +481,7 @@ def __getitem__(
481481
return (
482482
converters,
483483
calling_convention,
484-
candidate.requires_output_allocator,
484+
False,
485485
)
486486

487487
raise KeyError(

py/torch_tensorrt/dynamo/lowering/passes/remove_num_users_is_0_nodes.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@ def remove_num_users_is_0_nodes(
1616
output_node = list(gm.graph.nodes)[-1]
1717

1818
for node in gm.graph.nodes:
19-
if node != output_node and len(node.users) == 0:
19+
if (
20+
node != output_node
21+
and len(node.users) == 0
22+
and len(node.all_input_nodes) > 0
23+
):
2024
node_input = node.all_input_nodes[0]
2125
node.replace_all_uses_with(node_input)
2226
gm.graph.erase_node(node)

0 commit comments

Comments
 (0)