Merge branch 'pytorch:main' into bm/T145005253/build-scripts-8

pytorch · Mar 14, 2023 · 65d8fca · 65d8fca
2 parents c0ab4d1 + f388b95
commit 65d8fca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fbgemm_gpu/src/jagged_tensor_ops_autograd.cpp b/fbgemm_gpu/src/jagged_tensor_ops_autograd.cpp
@@ -644,7 +644,7 @@ jagged_dense_elementwise_add_jagged_output(
     const Tensor& y) {
   // Convert to jagged
   auto jagged_values =
-      DenseToJaggedOp::apply(y, x_offsets, c10::optional<int64_t>())[0];
+      DenseToJaggedOp::apply(y, x_offsets, x_values.size(0))[0];
 
   // Add jagged_values + x_values -> sum_values
   auto sum_values = x_values + jagged_values;