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

fix(docs): typos in macro name

Browse Source 
chore(docs): sync .md file
This commit is contained in:
Jérome Eertmans 2024-11-19 16:42:19 +01:00
parent f7ae0f99fe
commit d912034cb5
No known key found for this signature in database
GPG Key ID: E540D84E8C113AB1
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/ffi.ipynb
View File

@ -139,8 +139,8 @@
"}\n",
"\n",
"// Wrap `RmsNormImpl` and specify the interface to XLA. If you need to declare\n",
"// this handler in a header, you can use the `XLA_FFI_DECLASE_HANDLER_SYMBOL`\n",
"// macro: `XLA_FFI_DECLASE_HANDLER_SYMBOL(RmsNorm)`.\n",
"// this handler in a header, you can use the `XLA_FFI_DECLARE_HANDLER_SYMBOL`\n",
"// macro: `XLA_FFI_DECLARE_HANDLER_SYMBOL(RmsNorm)`.\n",
"XLA_FFI_DEFINE_HANDLER_SYMBOL(\n",
" RmsNorm, RmsNormImpl,\n",
" ffi::Ffi::Bind()\n",

4
docs/ffi.md
View File

@ -134,8 +134,8 @@ ffi::Error RmsNormImpl(float eps, ffi::Buffer<ffi::F32> 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()