George Hotz f7291f6ca3 fixes big KOPT, breaks opencl (#505) ...

* fixes big KOPT, breaks opencl

* fix optimizer

* KernelCache

* oops, broke batchnorm

* hack to fix it

* fix llvm, less hacky gpu

* disable the cache

* cache just breaks things

2023-02-05 10:46:17 -08:00

..

mnist

move datasets into datasets

2021-10-30 19:55:50 -07:00

__init__.py

fixes big KOPT, breaks opencl (#505)

2023-02-05 10:46:17 -08:00

imagenet.py

fix bug in gpu copy out

2023-01-30 16:51:28 -08:00