[codemod] Fix signed shift error in caffe2/torch/fb/retrieval/fused_kmean_ann_jagged_output.cu

Summary:
Bit-shifting signed variables is scary because the bit-shift can result in the signed variable becoming negative, which is often unwanted.

This diff fixes such a situation, which is often as a simple as changing `1 << 31` to `1ul << 31` or an equivalent construction.

 - If you approve of this diff, please use the "Accept & Ship" button :-)

Reviewed By: wenxin0319

Differential Revision: D72536140

fbshipit-source-id: 887255469368f5b4b500dbb42ff7f3307e6f68bb

Author: r-barnes

torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu

@@ -241,11 +241,11 @@ void deformable_im2col(
   // https://github.com/pytorch/vision/issues/4269
   bool use_64bits_indexing = false;
   // Checks if num_kernels or columns numel larger than 2 ** 31
-  use_64bits_indexing |= num_kernels > (1 << 31);
+  use_64bits_indexing |= num_kernels > std::numeric_limits<int32_t>::max();
   use_64bits_indexing |=
       ((int64_t)n_in_channels * weight_h * weight_w * parallel_imgs * out_h *
            out_w >
-       (1 << 31));
+           std::numeric_limits<int32_t>::max());
 
   if (use_64bits_indexing) {
     AT_DISPATCH_FLOATING_TYPES_AND_HALF(
@@ -432,7 +432,7 @@ void compute_grad_input(
   // https://github.com/pytorch/vision/issues/4269
   bool use_64bits_indexing = false;
   // Checks if num_kernels or columns numel larger than 2 ** 31
-  use_64bits_indexing |= num_kernels > (1 << 31);
+  use_64bits_indexing |= num_kernels > std::numeric_limits<int32_t>::max();
 
   at::globalContext().alertNotDeterministic("compute_grad_input");
 
@@ -673,10 +673,10 @@ void compute_grad_offset_and_mask(
   // https://github.com/pytorch/vision/issues/4269
   bool use_64bits_indexing = false;
   // Checks if columns numel is larger than 2 ** 31
-  use_64bits_indexing |= num_kernels > (1 << 31);
+  use_64bits_indexing |= num_kernels > std::numeric_limits<int32_t>::max();
   use_64bits_indexing |=
       ((int64_t)channels * weight_h * weight_w * parallel_imgs * out_h * out_w >
-       (1 << 31));
+       std::numeric_limits<int32_t>::max());
 
   if (use_64bits_indexing) {
     AT_DISPATCH_FLOATING_TYPES_AND_HALF(