Commit Graph
19 Commits
Author SHA1 Message Date
Sidharth N. BabuandGitHub ef14dfb277 compile fixes (#10442) 2025-06-06 18:38:37 -04:00
George HotzandGitHub d71fe7faa5 rename allocator methods to not conflict [pr] (#7788)
* rename allocator methods to not conflict [pr]

* forgot those

* transfer + offset
2024-11-20 00:10:29 +08:00
George HotzandGitHub 150ea2eb76 create engine folder and move code (#3948)
* retry

* older tf

* that
2024-03-26 20:38:03 -07:00
George HotzandGitHub 41efaa848c move graph.py and jit.py into features (#3376)
* move graph.py into features

* move jit into features

* fix quickstart
2024-02-12 17:34:34 +01:00
Yoshinori SanoandGitHub 98c732cf9d fix metal compile error in extra/gemm (#3365) 2024-02-10 12:54:41 +01:00
George HotzandGitHub a280cfe169 move dtypes to dtype.py (#2964)
* move dtypes to dtype.py

* fix urllib
2024-01-01 14:58:48 -08:00
f409b57854 update metal matmul and matvec for new device style (#2732)
* update for new device style

* create device before compile

---------

Co-authored-by: chenyu <[email protected]>
2023-12-17 16:15:07 -05:00
Rory ClearandGitHub 553688f12a update metal matmul and matvec for compile api (#2238) 2023-11-08 08:08:35 -08:00
George HotzandGitHub 67e34b356a good stuff from tensor cores branch (#1199) 2023-07-08 16:58:26 -07:00
geohot 8b777af571 metal_conv gets over 10.4 TFLOPS... 2023-04-15 03:31:22 -07:00
geohot d66e682205 metal matmul from tcores branch 2023-04-14 23:29:29 -07:00
geohot 68e45fca18 metal_matmul: bw and torch sync 2023-03-23 08:02:04 -07:00
geohot bd6c3c31a9 compare to torch 2023-03-22 23:58:37 -07:00
geohot c3a3db75c7 fix metal matmul example 2023-03-22 23:42:51 -07:00
George HotzandGitHub 1a039306d2 good changes from llama branch (#671)
* good changes from llama

* transpose behavior changed
2023-03-09 20:51:22 -08:00
George HotzandGitHub bfcec234a2 Refactor ASTs (#622)
* ugh worst branch name

* compiler refactor continues

* scc -> cloc

* buf -> _buf

* finish _buf, and program -> runtime

* gpu is still working, clang isn't

* clang in new style

* ops_metal

* something broke it

* improve metal

* clean up tons of cl crap

* hack fix sync

* cleaner gpu

* gpu metal clang

* cleanups

* minor refactor

* GPUCodegen

* fix up LLVM

* blind CUDA refactor

* codegen / runtime

* keep ops naming

* linter passes

* woah, llvm was allocing 4x what it needed to

* bugfixes

* fix openpilot compiler

* fix compile_efficientnet

* method cache should fix tests

* deal with duped functions
2023-03-01 18:57:29 -08:00
calleditandGitHub 81f7c6800a Added info on simdgroup availability (#586)
* Add info on simdgroup availability

* "osx" not "os x"

* Update metal_matmul.py

* Update metal_matmul.py
2023-02-23 13:59:02 -08:00
geohot bbfec2fde7 8.46 TFLOPS 2023-02-19 13:21:25 -08:00
geohot 1ba847963d reshape and retain metal_matmul 2023-02-19 13:07:23 -08:00