Custom cuda kernel (#2517)

2025-09-27 08:18:30 +08:00 · 2025-08-20 17:20:22 -07:00
parent f4c8888cbe
commit e397177f6e
19 changed files with 1042 additions and 211 deletions
--- a/docs/src/python/cuda.rst
+++ b/docs/src/python/cuda.rst
@@ -0,0 +1,9 @@
+CUDA
+=====
+
+.. currentmodule:: mlx.core.cuda
+
+.. autosummary::
+  :toctree: _autosummary
+
+  is_available
--- a/docs/src/python/fast.rst
+++ b/docs/src/python/fast.rst
@@ -13,3 +13,4 @@ Fast
  rope
  scaled_dot_product_attention
  metal_kernel
+  cuda_kernel