diff --git a/docs/ffi/rms_norm.cc b/docs/ffi/rms_norm.cc index 467f13d44..335c81d33 100644 --- a/docs/ffi/rms_norm.cc +++ b/docs/ffi/rms_norm.cc @@ -69,8 +69,8 @@ ffi::Error RmsNormImpl(float eps, ffi::Buffer x, } // Wrap `RmsNormImpl` and specify the interface to XLA. If you need to declare -// this handler in a header, you can use the `XLA_FFI_DECLASE_HANDLER_SYMBOL` -// macro: `XLA_FFI_DECLASE_HANDLER_SYMBOL(RmsNorm)`. +// this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL` +// macro: `XLA_FFI_DECLARE_HANDLER_SYMBOL(RmsNorm)`. XLA_FFI_DEFINE_HANDLER_SYMBOL(RmsNorm, RmsNormImpl, ffi::Ffi::Bind() .Attr("eps") diff --git a/examples/ffi/src/jax_ffi_example/rms_norm.cc b/examples/ffi/src/jax_ffi_example/rms_norm.cc index b24f61301..8314219c4 100644 --- a/examples/ffi/src/jax_ffi_example/rms_norm.cc +++ b/examples/ffi/src/jax_ffi_example/rms_norm.cc @@ -71,8 +71,8 @@ ffi::Error RmsNormImpl(float eps, ffi::Buffer x, } // Wrap `RmsNormImpl` and specify the interface to XLA. If you need to declare -// this handler in a header, you can use the `XLA_FFI_DECLASE_HANDLER_SYMBOL` -// macro: `XLA_FFI_DECLASE_HANDLER_SYMBOL(RmsNorm)`. +// this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL` +// macro: `XLA_FFI_DECLARE_HANDLER_SYMBOL(RmsNorm)`. XLA_FFI_DEFINE_HANDLER_SYMBOL(RmsNorm, RmsNormImpl, ffi::Ffi::Bind() .Attr("eps")