Skip to content

Commit 7390e7d

Browse files
authored
neg converter correction (#2307)
1 parent ea5a289 commit 7390e7d

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -304,18 +304,12 @@ def aten_ops_neg(
304304
kwargs: Dict[str, Argument],
305305
name: str,
306306
) -> Union[TRTTensor, Sequence[TRTTensor]]:
307-
input_val = args[0]
308-
if (isinstance(input_val, TRTTensor)) and (
309-
input_val.dtype == trt.int8 or input_val.dtype == trt.int32
310-
):
311-
input_val = cast_trt_tensor(network, input_val, trt.float32, name)
312-
313307
return impl.unary.neg(
314308
network,
315309
target,
316310
SourceIR.ATEN,
317311
name,
318-
input_val,
312+
args[0],
319313
)
320314

321315

0 commit comments

Comments
 (0)