5 Commits

Author	SHA1	Message	Date
Awni Hannun	1ca616844b	Fix unintuitive metal kernel caching (#2242) ... * Fix unintuitive metal kernel caching * alternative solution	2025-06-06 20:08:15 -07:00
Alex Barron	9e516b71ea	Add dispatchThreads to custom kernel doc (#1551) ... * add dispatchThreads info * update * add link	2024-11-01 13:07:48 -07:00
Awni Hannun	ba3e913c7a	Simplifications for MLX C (#1396) ... * simplifications for MLX C * use vectors instead of map * update examples	2024-09-06 19:16:50 -07:00
Alex Barron	b96e105244	Add grid_sample example to metal_kernel docs (#1352) ... * Add `zero_outputs` and `atomic_outputs` options to `metal_kernel` * add grid sample to docs * zero_outputs -> init_value * add missing header for linux	2024-08-23 18:24:16 -07:00
Alex Barron	0fd2a1f4b0	Custom Metal Kernels from Python (#1325) ... * start * simple kernels working * restructure * inverse example working * docs + fixes * missing file * fix imports * address comments * add docs + fix test * Review comments + refactor to a single function * update docs * remove hashing * fix contig bug in test * back to a class * trailing whitespace * fix tests * match c++ and python apis * add link + make args kw_only	2024-08-22 13:46:29 -07:00