From 5de272bee424ff59f8d69bd197256fe058ac3e41 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Mon, 10 Nov 2025 22:12:52 -0800 Subject: [PATCH] fix aoti graph break --- .../dynamo/runtime/meta_ops/register_meta_ops.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/py/torch_tensorrt/dynamo/runtime/meta_ops/register_meta_ops.py b/py/torch_tensorrt/dynamo/runtime/meta_ops/register_meta_ops.py index b0d1c69916..b5fc3d48fa 100644 --- a/py/torch_tensorrt/dynamo/runtime/meta_ops/register_meta_ops.py +++ b/py/torch_tensorrt/dynamo/runtime/meta_ops/register_meta_ops.py @@ -68,11 +68,13 @@ def fake_tensorrt_execute_engine( output_shape.append(min_val) else: output_shape.extend(outputs_mode_dict["opt"][out_idx].size()) - fake_outputs.append( - torch.empty(output_shape, dtype=outputs_mode_dict["opt"][out_idx].dtype) + torch.empty( + output_shape, + dtype=outputs_mode_dict["opt"][out_idx].dtype, + device=inputs[0].device, + ) ) - return fake_outputs