mlx/mlx/backend/metal/jit
Awni Hannun 40c62c1321
Use int64 stride everywhere (#1671)
* use int64 stride everywhere

* fix ext

* fix ext

* more shape + cleanup

* one more

* few more
2024-12-09 11:09:02 -08:00
..
arange.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
gemv_masked.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
includes.h Add gemv masked to JIT plus some fixes (#1310) 2024-08-07 13:38:07 -07:00
indexing.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
softmax.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
steel_conv.h Option to JIT steel gemm / conv (#1139) 2024-05-23 18:07:34 -07:00
steel_gemm.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00