From f039aa2a92dd053618f97e0357d25840006dd72f Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 30 Oct 2024 18:12:19 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- transformer_engine/jax/csrc/extensions/ffi.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/transformer_engine/jax/csrc/extensions/ffi.h b/transformer_engine/jax/csrc/extensions/ffi.h index 5efc507a90..48b6f69ea8 100644 --- a/transformer_engine/jax/csrc/extensions/ffi.h +++ b/transformer_engine/jax/csrc/extensions/ffi.h @@ -70,7 +70,8 @@ T get_attr_value(Dictionary& attrs, std::string attr_name, return attr.value(); } -inline size_t product(const xla::ffi::Span& data, size_t start_idx = 0, size_t end_idx = 0) { +inline size_t product(const xla::ffi::Span& data, size_t start_idx = 0, + size_t end_idx = 0) { end_idx = (end_idx == 0) ? data.size() : end_idx; return std::accumulate(data.begin() + start_idx, data.begin() + end_idx, size_t(1), std::multiplies());