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