More fixes for arrays with large sizes (#1405)

* compile works for big arrays when contiguous * style * nits in docs * a bunch more stuff * update jit * update jit * use constant for shapes and strides and remove elem_to_loc overload * use kernel instantiation * docs nits * update binary and ternary * comments
2025-12-16 01:49:05 +08:00 · 2024-09-17 12:46:31 -07:00
parent c6739ba7f3
commit 4f46e9c997
26 changed files with 325 additions and 611 deletions
--- a/docs/src/python/nn/functions.rst
+++ b/docs/src/python/nn/functions.rst
@@ -13,6 +13,7 @@ simple functions.
   :template: nn-module-template.rst

   elu
+   celu
   gelu
   gelu_approx
   gelu_fast_approx