github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-17 10:02:00 -05:00
Code Issues Packages Projects Releases Wiki Activity
2,587 Commits 227 Branches 14 Tags
e93e240a6c41ef57ef0e046050f187ff1aee7b2b
Commit Graph

4 Commits

Author SHA1 Message Date
nimlgen
1c0449e190 add cache collector (#1595) 
* init cache collector

* add test_cache_collector.py

* switch GlobalCounters.cache to CacheCollector

* init jit models test

* jitted SD

* add debug msg to print loaded bufs count

* moved cache collctor to jit

* clearer SD

* no double device import
 2023-08-28 19:59:55 -07:00
George Hotz
a6d842af7a move device to ops (#1646) 
* move device to ops

* mlops types

* 2 lines
 2023-08-23 08:30:17 -07:00
chenyu
ae39cf84ab Symbolic Shape JIT main PR (#1353) 
* Symbolic Shape JIT

update tests

2 variables symbolic ops, adding more tests

test passing

cleanup

* more test cases

* single flag

* review update

* jit attention one piece

* realize

* symbolic_jit test for cuda

* old artifact

* works with cuda gpu but failed ci

* CUDACPU
 2023-08-18 14:39:55 -07:00
George Hotz
20894991ed good changes from the M1 Tensor Core project (#730) 
* good changes

* working except llvm

* llvm types

* nice acc

* archprobe

* lang.float4

* use self.acc for late acc

* fix store bug
 2023-03-29 05:11:02 +04:00