changed from long to int64_t

Author: Chillee

functorch/csrc/BatchRulesConvolution.cpp

@@ -403,7 +403,7 @@ convolution_backward_weight_batch_rule(
 
 std::tuple<Tensor,Tensor,Tensor> convolution_backward_plumbing(
     const Tensor& grad_output_, const Tensor& input_, const Tensor& weight_,
-    const c10::OptionalArrayRef<long> bias_sizes_opt,
+    const c10::OptionalArrayRef<int64_t> bias_sizes_opt,
     IntArrayRef stride, IntArrayRef padding, IntArrayRef dilation, bool transposed,
     IntArrayRef output_padding, int64_t groups, std::array<bool, 3> output_mask) {
   const auto maybe_layer = maybeCurrentDynamicLayer();

functorch/csrc/BatchRulesModules.cpp

@@ -295,7 +295,7 @@ template <typename F, F Func, typename A, typename B, typename C, typename... T>
 struct UpsampleBackwardBatchRuleHelper<F, Func, typelist<A, B, C, T...>> {
   static std::tuple<Tensor,optional<int64_t>> apply(
       const Tensor& grad_output, optional<int64_t> grad_output_bdim,
-      OptionalArrayRef<long> output_size, IntArrayRef input_size,
+      OptionalArrayRef<int64_t> output_size, IntArrayRef input_size,
       T... extra_args) {
     auto grad_output_ = reshape_dim_into(*grad_output_bdim, 0, grad_output);
     TORCH_INTERNAL_ASSERT(input_size.size() > 0);