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 90491d7 commit 21c43f8Copy full SHA for 21c43f8
py/torch_tensorrt/dynamo/runtime/meta_ops/register_meta_ops.py
@@ -68,11 +68,13 @@ def fake_tensorrt_execute_engine(
68
output_shape.append(min_val)
69
else:
70
output_shape.extend(outputs_mode_dict["opt"][out_idx].size())
71
-
72
fake_outputs.append(
73
- torch.empty(output_shape, dtype=outputs_mode_dict["opt"][out_idx].dtype)
+ torch.empty(
+ output_shape,
74
+ dtype=outputs_mode_dict["opt"][out_idx].dtype,
75
+ device=inputs[0].device,
76
+ )
77
)
78
return fake_outputs
79
80
0 commit comments