clean up files

pytorch · Dec 18, 2024 · 7bce886 · 7bce886
1 parent 7290336
commit 7bce886
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/test/cpp/run_tests.sh b/test/cpp/run_tests.sh
@@ -105,9 +105,9 @@ fi
 for name in "${test_names[@]}"; do
   echo "Running $name cpp test..."
   if [ "$LOGFILE" != "" ]; then
-    bazel $BAZEL_VERB $EXTRA_FLAGS //torch_xla/csrc/runtime:all //test/cpp:${name} --test_timeout 1200 --test_output=all ${FILTER:+"$FILTER"} 2> $LOGFILE
+    bazel $BAZEL_VERB $EXTRA_FLAGS //torch_xla/csrc/runtime:all //test/cpp:${name} --test_timeout 1200 ${FILTER:+"$FILTER"} 2> $LOGFILE
   else
-    bazel $BAZEL_VERB $EXTRA_FLAGS //torch_xla/csrc/runtime:all //test/cpp:${name} --test_timeout 1200 --test_output=all ${FILTER:+"$FILTER"}
+    bazel $BAZEL_VERB $EXTRA_FLAGS //torch_xla/csrc/runtime:all //test/cpp:${name} --test_timeout 1200 ${FILTER:+"$FILTER"}
   fi
 done
 
diff --git a/torch_xla/csrc/reduction.h b/torch_xla/csrc/reduction.h
@@ -88,8 +88,9 @@ xla::XlaOp BuildCumulativeComputation(xla::XlaOp input, int64_t dim,
                                       const xla::XlaComputation& reducer,
                                       xla::XlaOp init);
 
-// Compute the cumulative computation specified by "reducer" and "init" in the
+// Computes the cumulative computation specified by "reducer" and "init" in the
 // given dimension "dim".
+// Returns a tuple XlaOp (values, indices).
 xla::XlaOp BuildCumulativeComputationWithIndices(
     xla::XlaOp value_input, xla::XlaOp index_input, int64_t dim,
     const xla::XlaComputation& reducer, xla::XlaOp value_init,