Removing cast

Author: apbose

py/torch_tensorrt/dynamo/conversion/converter_utils.py

@@ -1,7 +1,7 @@
 import functools
 import logging
 import re
-from typing import Any, List, Optional, Tuple, Union
+from typing import Any, Callable, List, Optional, Tuple, Union
 
 import numpy as np
 import tensorrt as trt
@@ -55,7 +55,9 @@ def dynamic_unsupported_with_args(
     arg_positions_to_check: Optional[List[int]] = None,
 ) -> Callable[[torch.fx.Node], bool]:
     """Returns a validator that a node has no dynamic args at specific positions"""
-    return functools.partial(_dynamic_unsupported, arg_positions_to_check=arg_positions_to_check)
+    return functools.partial(
+        _dynamic_unsupported, arg_positions_to_check=arg_positions_to_check
+    )
 
 
 def _dynamic_unsupported(

py/torch_tensorrt/dynamo/conversion/impl/split.py

@@ -28,18 +28,17 @@ def split(
             f"split received input {input} that is not part " "of the TensorRT region!"
         )
 
-    dim = cast(int, dim)
     dynamic_shape = has_dynamic_shape(input.shape)
     if dynamic_shape > 0:
         # Check whether slice target dim is dynamic shape dim
         assert input.shape[dim] != -1, "Can't chunk on dynamic shape dimension!"
 
     split_sizes = []
     if isinstance(split_size_or_sections, int):
-        split_sizes.append(cast(int, split_size_or_sections))
+        split_sizes.append(split_size_or_sections)
     else:
         for split_size_or_section in split_size_or_sections:
-            split_sizes.append(cast(int, split_size_or_section))
+            split_sizes.append(split_size_or_section)
 
     start = [0] * len(input.shape)
     stride = [1] * len(start)
@@ -65,7 +64,7 @@ def split(
     output = []
     for i in range(num_splits):
         shape = list(input.shape)
-        shape[dim] = min(split_sizes[i], cast(int, max_offset - offset))
+        shape[dim] = min(split_sizes[i], max_offset - offset)
         start[dim] = offset
         if dynamic_shape:
             shape = get_shape_with_dynamic_shape(