Skip to content

Commit 6afb8be

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 68e5a47 commit 6afb8be

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

transformer_engine/jax/csrc/extensions/inspect.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -121,14 +121,14 @@ Error_Type InspectFFI(cudaStream_t stream, Buffer_Type input_buf, Buffer_Type mi
121121

122122
XLA_FFI_DEFINE_HANDLER_SYMBOL(InspectHandler, InspectFFI,
123123
FFI::Bind()
124-
.Ctx<FFI_Stream_Type>() // stream
125-
.Arg<Buffer_Type>() // input
126-
.Arg<Buffer_Type>() // min
127-
.Arg<Buffer_Type>() // max
128-
.Arg<Buffer_Type>() // mean
129-
.Arg<Buffer_Type>() // std
130-
.Ret<Buffer_Type>() // output
131-
.Attr<std::string_view>("name") // probe name
124+
.Ctx<FFI_Stream_Type>() // stream
125+
.Arg<Buffer_Type>() // input
126+
.Arg<Buffer_Type>() // min
127+
.Arg<Buffer_Type>() // max
128+
.Arg<Buffer_Type>() // mean
129+
.Arg<Buffer_Type>() // std
130+
.Ret<Buffer_Type>() // output
131+
.Attr<std::string_view>("name") // probe name
132132
);
133133

134134
} // namespace jax

0 commit comments

Comments
 (0)