Commit graph

10 commits

Author SHA1 Message Date
George Hotz
ac079e43d7
ElementwiseMixin (#14777) 2026-02-16 08:50:47 +08:00
qazal
33b31d9cd6
tinykittens flash attention dtype fix, add CI (#14770)
* don't hardcdoe amd device

* add failing tests, ci too

* fix: fix for dtype mixin

* bump to rocm 7.1

---------

Co-authored-by: Woze Parrot <wozeparrot@gmail.com>
2026-02-16 01:15:11 +09:00
wozeparrot
89c4206e22
fix: typing (#13614) 2025-12-07 20:10:30 -08:00
wozeparrot
62e2fc5108
tk: global load/store rv (#13577) 2025-12-04 17:23:48 -08:00
wozeparrot
ffc31a23f4
tk mi350 (#13288) 2025-11-25 15:49:44 -08:00
wozeparrot
56b2540349
tk: keep extra tile data by replacing uop (#13370) 2025-11-19 15:11:43 -08:00
wozeparrot
777cbec5b3
tk: rename rt tile dims to base (#13265) 2025-11-13 18:43:02 -08:00
wozeparrot
7eb0d8e744
feat: mixins on tiles (#13246) 2025-11-13 16:52:52 -08:00
wozeparrot
371c1f2355
tk: move tiles to class (#13224) 2025-11-11 21:53:46 -08:00
wozeparrot
6252831ceb
feat: initial tk library (#13160) 2025-11-09 22:54:29 -08:00