Skip to content

Commit 21c43f8

Browse files
fix aoti graph break (#3892)
1 parent 90491d7 commit 21c43f8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

py/torch_tensorrt/dynamo/runtime/meta_ops/register_meta_ops.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,11 +68,13 @@ def fake_tensorrt_execute_engine(
6868
output_shape.append(min_val)
6969
else:
7070
output_shape.extend(outputs_mode_dict["opt"][out_idx].size())
71-
7271
fake_outputs.append(
73-
torch.empty(output_shape, dtype=outputs_mode_dict["opt"][out_idx].dtype)
72+
torch.empty(
73+
output_shape,
74+
dtype=outputs_mode_dict["opt"][out_idx].dtype,
75+
device=inputs[0].device,
76+
)
7477
)
75-
7678
return fake_outputs
7779

7880

0 commit comments

Comments
 (0)