mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
13,471 commits 280 branches 15 tags 325 MiB
Commit graph

13,471 commits

This branch
This branch
All branches
Author SHA1 Message Date
Sohaib
8835df7a5c
 		upgrade onnx to 1.13.0 (#588) 
- remove protobuf from direct dependencies
- replace deprecated mapping.TENSOR_TYPE_TO_NP_TYPE

Co-authored-by: Sohaib Errabii <sohaib.errabii@ipops.io>
 2023-02-23 13:59:23 -08:00
calledit
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
George Hotz
d22e19536b onnx: support low quality Resize. stuck on ConvTranspose will have to wait for convless 2023-02-23 09:05:23 -08:00
George Hotz
ab3a2ae9a2 fix test_resnet in onnx now that maxpool works 2023-02-23 08:41:47 -08:00
George Hotz
fd6082dcef support all _pool2d. conv will eventually be an hlop 2023-02-23 08:19:47 -08:00
George Hotz
94ccab941e compile_tensorflow: no cast required 2023-02-22 21:14:21 -08:00
George Hotz
135d0ddb78 compile_tensorflow: read weights from disk 2023-02-22 21:12:35 -08:00
George Hotz
0615dcffe7 compile_tensorflow: save the weights 2023-02-22 21:05:45 -08:00
George Hotz
c537fd0614 compile_tensorflow: add initialize and tests 2023-02-22 20:50:53 -08:00
George Hotz
dc914cde50 compile_tensorflow 2023-02-22 20:08:58 -08:00
George Hotz
10ba96029c device: ImportError -> Exception 2023-02-22 19:48:59 -08:00
George Hotz
76b4d0577d yolov8 works up to the MaxPool 2023-02-22 19:32:13 -08:00
Mischa Untaga
5190784cbb
 		Fix Tensor random functions determinism with same seed (#580) 
* fix Tensor random functions determinism with same seed

* long lived rng

* TIL ClassVar typing
 2023-02-22 19:08:43 -08:00
George Hotz
c8d89eb20e avg/max pool strides 2023-02-22 18:00:48 -08:00
George Hotz
ac8daaeea5 refactor device, credit martinloretzzz 2023-02-22 17:29:40 -08:00
George Hotz
a3ddc1d484 CLANG doesn't support float4 2023-02-22 10:41:15 -08:00
George Hotz
c4c2c28738
 		a sustainable approach to float4 (#582) 
* a sustainable approach to float4

* can_float4

* fix tests

* fix float4

* delete dead code

* types and minor cleanup
 2023-02-22 09:45:08 -08:00
George Hotz
50e541174c fix image copyout 2023-02-22 07:27:13 -08:00
George Hotz
628ce067a1 add tests to mypy 2023-02-22 07:07:38 -08:00
George Hotz
104c3c5e73 oops, forgot that debug 2023-02-22 06:58:27 -08:00
George Hotz
c5e2126d49 move DEBUG to helpers 2023-02-22 06:52:11 -08:00
Mischa Untaga
14bb2c40a2
 		Fix yolov3 example (#577) 2023-02-21 09:24:00 -08:00
Martin Loretz
8550b3e168
 		Fix kernel cache key (#570) 2023-02-21 03:53:07 -08:00
George Hotz
66b4b3bdd3 clang: linux needs fPIC 2023-02-20 21:09:33 -08:00
George Hotz
4f4b7d05a7 clang needs -lm and is very slow 2023-02-20 21:02:04 -08:00
George Hotz
4126bf2982 remove six (hopefully not needed) 2023-02-20 20:44:23 -08:00
George Hotz
efcb3f0cdd fix metal dep 2023-02-20 20:43:32 -08:00
Connor Henderson
9670bf1fd1
 		Add unsqueeze (#574) 
* Add unsqueeze

* remove UNSQUEEZE from llops part of readme

* make it an hlop
 2023-02-20 20:14:59 -08:00
George Hotz
cfad2902d5 clang: that might have been getting GCed 2023-02-20 20:13:05 -08:00
Diogo
506970414a
 		added metal packages to setup and release metal buffers after del (#571) 2023-02-20 20:01:07 -08:00
George Hotz
d9fa47ecc9 use the TinyJit in the efficientnet runner, 200ms -> 20ms 2023-02-20 19:58:16 -08:00
George Hotz
714bf4b108
 		clang backend (#572) 
* start clang backend

* mostly working

* no group for reduce w clang

* it compiles

* compiles

* a11y

* minor fixups

* formatting

* add a test

* rename test
 2023-02-20 18:18:18 -08:00
George Hotz
353e81fa8f need hand opt with IMAGE==2 2023-02-20 10:18:19 -08:00
George Hotz
b925d05e22 no networkx is okay 2023-02-20 09:57:12 -08:00
George Hotz
4d232c7c95 optional networkx + DEBUGCL=2 2023-02-20 09:50:46 -08:00
George Hotz
8b0082540b openpilot compile cleanups 2023-02-20 09:16:03 -08:00
George Hotz
ea13504f35 fix METAL_XCODE 2023-02-19 20:02:12 -08:00
George Hotz
bbfec2fde7 8.46 TFLOPS 2023-02-19 13:21:25 -08:00
George Hotz
1ba847963d reshape and retain metal_matmul 2023-02-19 13:07:23 -08:00
George Hotz
60008e55cd sick of that failing 2023-02-19 13:05:37 -08:00
George Hotz
0b3f686530
 		Good fast triton (#567) 
* runtime fixups

* uints and printbufs

* uints don't work
 2023-02-19 12:21:55 -08:00
Jacky Lee
cb679cd051
 		Fix weight initialization (#566) 
* Fix weight initialization

* Use scaled_uniform in serious_mnist
 2023-02-19 11:25:29 -08:00
Diogo
a508c2b429
 		small tweaks to the metal runtime (#562) 
* small tweaks to the metal runtime

* create buffer straight from numpy

* reverted back due to bug when adding 1+1

* removed comments
 2023-02-19 11:25:13 -08:00
Martin Loretz
7e9a5e3f31
 		Refactor graph (#560) 
* Refactor graph

* Add graph tests

* Use CPUBuffer for graph tests

* Remove the use of GlobalCounters
 2023-02-19 10:41:30 -08:00
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