mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-26 05:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix typos

Browse Source
This commit is contained in:
rajasekharporeddy 2025-02-18 17:01:29 +05:30
parent e78a469b42
commit 180be99798
2 changed files with 4 additions and 4 deletions
docs/ffi
rms_norm.cc
examples/ffi/src/jax_ffi_example
rms_norm.cc

4
docs/ffi/rms_norm.cc

@ -69,8 +69,8 @@ ffi::Error RmsNormImpl(float eps, ffi::Buffer<ffi::F32> x,
} }
// Wrap `RmsNormImpl` and specify the interface to XLA. If you need to declare // 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` // this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL`
// macro: `XLA_FFI_DECLASE_HANDLER_SYMBOL(RmsNorm)`. // macro: `XLA_FFI_DECLARE_HANDLER_SYMBOL(RmsNorm)`.
XLA_FFI_DEFINE_HANDLER_SYMBOL(RmsNorm, RmsNormImpl, XLA_FFI_DEFINE_HANDLER_SYMBOL(RmsNorm, RmsNormImpl,
ffi::Ffi::Bind() ffi::Ffi::Bind()
.Attr<float>("eps") .Attr<float>("eps")

4
examples/ffi/src/jax_ffi_example/rms_norm.cc

@ -71,8 +71,8 @@ ffi::Error RmsNormImpl(float eps, ffi::Buffer<ffi::F32> x,
} }
// Wrap `RmsNormImpl` and specify the interface to XLA. If you need to declare // 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` // this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL`
// macro: `XLA_FFI_DECLASE_HANDLER_SYMBOL(RmsNorm)`. // macro: `XLA_FFI_DECLARE_HANDLER_SYMBOL(RmsNorm)`.
XLA_FFI_DEFINE_HANDLER_SYMBOL(RmsNorm, RmsNormImpl, XLA_FFI_DEFINE_HANDLER_SYMBOL(RmsNorm, RmsNormImpl,
ffi::Ffi::Bind() ffi::Ffi::Bind()
.Attr<float>("eps") .Attr<float>("eps")