delete int2float conversion for some ops

Author: zewenli98

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

@@ -253,12 +253,6 @@ def add(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network, target, source_ir, name, trt.ElementWiseOperation.SUM, lhs_val, rhs_val
     )
@@ -272,12 +266,6 @@ def mul(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network,
         target,
@@ -297,12 +285,6 @@ def max(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network, target, source_ir, name, trt.ElementWiseOperation.MAX, lhs_val, rhs_val
     )
@@ -316,12 +298,6 @@ def min(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network, target, source_ir, name, trt.ElementWiseOperation.MIN, lhs_val, rhs_val
     )
@@ -335,12 +311,6 @@ def sub(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network, target, source_ir, name, trt.ElementWiseOperation.SUB, lhs_val, rhs_val
     )
@@ -392,12 +362,6 @@ def floor_divide(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network,
         target,
@@ -474,12 +438,6 @@ def eq(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network,
         target,
@@ -499,12 +457,6 @@ def gt(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network,
         target,
@@ -524,12 +476,6 @@ def lt(
     lhs_val: Union[TRTTensor, int, float],
     rhs_val: Union[TRTTensor, int, float],
 ) -> TRTTensor:
-    if isinstance(lhs_val, TRTTensor):
-        lhs_val = trt_cast_int_to_float(network, name, lhs_val)
-
-    if isinstance(rhs_val, TRTTensor):
-        rhs_val = trt_cast_int_to_float(network, name, rhs_val)
-
     return convert_binary_elementwise(
         network,
         target,