Commit Graph

1477 Commits

Author SHA1 Message Date
Kirill
7944cfdadc Remove Tensor.data (#565) 2023-02-18 16:36:12 -08:00
George Hotz
4efe0169bb remove old metal implementation 2023-02-18 13:51:48 -08:00
Jacky Lee
7e8b0305f3 Fix mnist gan example (#563) 2023-02-18 13:45:37 -08:00
Jacky Lee
9fd41632c6 Import get_parameters from tinygrad.nn (#559)
* get_parameter is in optim

* Update all imports for get_parameters

* Clean up

* use optim.get_paramters
2023-02-17 15:22:26 -08:00
George Hotz
fae7654924 fix sync issue 2023-02-17 12:42:45 -08:00
George Hotz
5e6265be6e metal timing, fix speed test 2023-02-17 12:31:54 -08:00
George Hotz
121bd03cbd metal globalcounters 2023-02-17 12:02:54 -08:00
George Hotz
67d1df80ba gid is array, metal works 2023-02-17 11:54:50 -08:00
George Hotz
f9af0322e7 metal can add 2023-02-17 11:45:33 -08:00
Jacky Lee
e172f0087a BatchNorm2D -> BatchNorm2d (#558)
* BatchNorm2D -> BatchNorm2d

* Fix typo
2023-02-16 12:31:49 -08:00
Liam
09315ef34f Add tinygrad.org reference in Readme. (#556) 2023-02-14 09:39:00 -08:00
George Hotz
82c257e8f5 more kernel search 2023-02-12 10:34:56 -08:00
Jacky Lee
c35fcc6964 Replace phrase for prompt (#555) 2023-02-12 09:04:44 -08:00
George Hotz
20a03d5017 woah, don't sync torch if it's not torch 2023-02-12 07:48:56 -08:00
George Hotz
de71c13934 test speed v torch uses jit 2023-02-12 07:43:17 -08:00
George Hotz
693d4b89a4 fixup TRITON backend to use new APIs 2023-02-12 06:57:49 -08:00
George Hotz
191c76cfd7 hlb_cifar10 torch version 2023-02-11 18:04:40 -08:00
George Hotz
9057d98d36 no lr decay in cifar. test this in torch tomorrow 2023-02-11 17:42:54 -08:00
George Hotz
dd7accb9cc decay LR, little bugfix 2023-02-11 17:34:15 -08:00
George Hotz
ba3bf5bdf7 cifar stops learning 2023-02-11 17:21:42 -08:00
George Hotz
40f3949742 fancier KOPT 2023-02-11 16:40:25 -08:00
George Hotz
446442dbb3 fix tests symbolic 2023-02-11 15:16:47 -08:00
George Hotz
7a7046f264 sum_combine_num 2023-02-11 14:48:31 -08:00
Kirill
a4f5f2ff8b Add missing packages to setup.py (#554) 2023-02-11 14:41:56 -08:00
George Hotz
20a351a3c6 hand optim CONVW 2023-02-11 14:41:08 -08:00
George Hotz
89499b303d oops, bad else. why didn't linter catch 2023-02-11 12:02:09 -08:00
George Hotz
7d33f2d659 CL.CACHE is over, GlobalCounters.cache is it 2023-02-11 12:00:14 -08:00
George Hotz
b9eae94ae9 move Device back into lazy 2023-02-11 11:26:53 -08:00
George Hotz
9152bb5b4a momentum support in SGD 2023-02-11 10:22:37 -08:00
George Hotz
0a2035e015 oops, GPU isn't defined 2023-02-11 10:10:02 -08:00
George Hotz
3421d4af10 the jit has a test 2023-02-11 10:04:03 -08:00
George Hotz
031edd01e6 switch openpilot compile to TinyJit 2023-02-11 09:51:44 -08:00
jspieler
8f912c3966 added deep deterministic policy gradient example (#531) 2023-02-11 10:10:46 -06:00
George Hotz
608fd730d3 put the JIT in extra 2023-02-11 00:35:18 -06:00
George Hotz
ed8ae7522a tinyjit 2023-02-11 00:22:36 -06:00
George Hotz
4c90a15689 make the fake data actually learnable 2023-02-10 23:35:21 -06:00
George Hotz
07629d7476 fakedata and move to new cache 2023-02-10 23:32:31 -06:00
George Hotz
63fa7daf30 wrong place for CL 2023-02-10 23:22:24 -06:00
George Hotz
6f9b103878 fix opencl types 2023-02-10 23:18:39 -06:00
George Hotz
fed95119dc CL.mem_used -> GlobalCounters.mem_used 2023-02-10 23:13:29 -06:00
George Hotz
51037815b9 add comment so we don't remove self.t tensor again 2023-02-10 23:07:07 -06:00
George Hotz
c0ea538ba0 Revert "revert t as tensor, constant folding should be done better"
This reverts commit 1d800a94ad.
2023-02-10 23:06:00 -06:00
George Hotz
1d800a94ad revert t as tensor, constant folding should be done better 2023-02-10 22:58:39 -06:00
George Hotz
77988e3236 fix str() line count bug in scc 2023-02-10 22:53:30 -06:00
George Hotz
1fb5b8069b simpler processed check 2023-02-10 22:49:20 -06:00
George Hotz
609477656e clean up lazy processing_op 2023-02-10 22:41:52 -06:00
George Hotz
a4cb161bd4 log_kernel 2023-02-10 21:51:53 -06:00
George Hotz
b9f02671d3 oops, broke torch speed test 2023-02-10 16:13:53 -06:00
George Hotz
0efe1e435f no need to render to check valid 2023-02-10 15:35:12 -06:00
Kirill
27154db99a Downloads weights in examples/stable_diffusion.py (#537)
* Downloads weights in examples/stable_diffusion.py

* use download_file_if_not_exists in fetch

* make consistent with previous NOCACHE behavior
2023-02-10 14:37:04 -06:00