test

Author: lanluo-nvidia

py/torch_tensorrt/dynamo/conversion/impl/unsqueeze.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Sequence
+from typing import List, Optional, Sequence, cast
 
 from torch.fx.node import Target
 from torch_tensorrt.dynamo._SourceIR import SourceIR
@@ -41,6 +41,8 @@ def unsqueeze_old(
             "of the TensorRT region!"
         )
 
+    dim = cast(int, dim)
+
     input_shape_size = len(input_val.shape)
     dim = get_positive_dim(dim, input_shape_size + 1)