scatter CI failures (#2925)

Author: apbose

py/torch_tensorrt/dynamo/lowering/_decompositions.py

@@ -207,9 +207,10 @@ def slice_scatter_decomposition(
         if i != dim:
             index_tensor_shape.append(src_each_dim)
     for index in range(start, end, step):
-        cat_tensors.append(index * torch.ones(index_tensor_shape, dtype=torch.long))
+        cat_tensors.append(index * torch.ones(index_tensor_shape, dtype=torch.int64))
     index_tensor = torch.stack(cat_tensors, dim).cuda()
-    output_tensor = torch.scatter(input_tensor, dim, index_tensor, src_tensor)
+    index_tensor_64 = index_tensor.to(torch.int64)
+    output_tensor = torch.scatter(input_tensor, dim, index_tensor_64, src_tensor)
     return output_tensor