Skip to content

Commit 1e65a53

Browse files
committed
add output.dtype
1 parent e7ed7de commit 1e65a53

File tree

1 file changed

+1
-0
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/elementwise

1 file changed

+1
-0
lines changed

py/torch_tensorrt/dynamo/conversion/impl/elementwise/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,4 +177,5 @@ def convert_binary_elementwise(
177177
output = layer.get_output(0)
178178
kind: str = str(target.__name__) if callable(target) else target
179179
output.name = output.name + "_" + kind
180+
output.dtype
180181
return output

0 commit comments

Comments
 (0)