handling the case wjem stop is max int64 value in the dynamic dimension

Author: apbose

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

@@ -2,6 +2,7 @@
 from typing import Optional, Sequence
 
 import numpy as np
+import sys
 import tensorrt as trt
 from torch.fx.node import Target
 from torch_tensorrt.dynamo._SourceIR import SourceIR
@@ -65,15 +66,16 @@ def slice_op(  # TODO: This should be slice not whatever is in base
         start_slice[dim] = start
     else:
         # the start and stop or None is dynamic along dim
-        if start < 0 or stop < 0 or stop_dynamic_None:
+        if start < 0 or stop < 0 or stop_dynamic_None or stop == sys.maxsize:
             # special assignments for dynamic cases
             if start < 0:
                 start_slice = input.shape
                 start_slice[dim] = -1 * start
             if stop < 0 or stop_dynamic_None:
                 stop_slice = [0] * len(input.shape)
                 stop_slice[dim] = -1 * stop
-
+            if(stop == sys.maxsize):
+                stop_slice = [0] * len(input.shape)
             start_slice_tensor = cat(
                 ctx,
                 target,
@@ -115,7 +117,7 @@ def slice_op(  # TODO: This should be slice not whatever is in base
                     shape,
                     start_slice_tensor,
                 )
-            if (stop < 0) or stop_dynamic_None:
+            if ((stop < 0) or stop_dynamic_None or stop == sys.maxsize):
                 shape = get_shape_with_dynamic_shape(
                     ctx, target, source_ir, name, output_shape, input
                 )