github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-15 01:48:23 -05:00
Code Issues Packages Projects Releases Wiki Activity
1,581 Commits 204 Branches 14 Tags
2f17d151b33dee7672e535fcbb930bfdc369bb96
Commit Graph

1581 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
George Hotz
2f17d151b3 fix batchnorm not realizing 2023-02-27 10:19:54 -08:00
George Hotz
c9252d38b2 mypy cache breaks if you sometimes check untyped defs, no checking tests for now 2023-02-27 09:57:33 -08:00
George Hotz
e74779f19d typing fixup 2023-02-27 09:52:04 -08:00
George Hotz
9aaa7edd74 no rnn support in onnx 2023-02-27 08:25:48 -08:00
Diogo
07e643431c added onnx group norm (#614) 2023-02-27 08:11:01 -08:00
George Hotz
edc8fbfff2 woah, why isn't OPT=2 2023-02-27 08:03:31 -08:00
George Hotz
a52913b242 test conv shapetracker has one view 2023-02-27 07:54:47 -08:00
George Hotz
57aec11736 minor randomness touchup 2023-02-27 07:06:26 -08:00
George Hotz
f10ccf7ec1 fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
George Hotz
686a74de92 fast zeros and ones 2023-02-27 06:46:26 -08:00
Diogo
e68fa18c9b layer norm support in onnx (#607) 
* layer norm support

* switched to 1e-05
 2023-02-26 22:04:02 -08:00
George Hotz
c4856aa193 fix yolo webcam 2023-02-26 17:24:05 -08:00
George Hotz
3a2a500e90 prevent race condition, external yolo test for now 2023-02-26 17:08:24 -08:00
Jacky Lee
0f58c4c648 Cleanup yolo and remove stateless classes (#604) 
* Add AvgPool2d as a layer

* Clean up a bit

* Remove stateless layers in yolo_nn

* More cleanup

* Save label for test

* Add test for YOLO

* Test without cv2

* Don't fail if cv2 not installed

* Better import

* Fix image read

* Use opencv :)

* Don't download the file

* Fix errors

* Use same version

* Set higher confidence

* Why is the confidence so low?

* Start over

* Remove stateless layers

* Remove extra lines

* Revert changes

* Save a few more lines
 2023-02-26 16:55:21 -08:00
Sohaib
71ae6e5605 fix: avgpool without counting padding (#605) 2023-02-26 07:13:00 -08:00
George Hotz
2b6d8e788d pushing permutes is done 2023-02-25 21:37:46 -08:00
George Hotz
f4ee7d2cad back to 196 kernels 2023-02-25 18:25:34 -08:00
George Hotz
187738f64f test fix, wow fixing contig exposed so much 2023-02-25 18:01:14 -08:00
George Hotz
c3d19101cb more debug info 2023-02-25 17:52:46 -08:00
George Hotz
6e98a172a0 fix broken contiguous 2023-02-25 17:41:49 -08:00
George Hotz
a44e8e4385 discard children on mop shuffle, 200 -> 196 kernels 2023-02-25 10:51:07 -08:00
George Hotz
1d01842232 remove fake test 2023-02-25 10:21:07 -08:00
George Hotz
8b96522e1d instant identity removal 2023-02-25 09:46:04 -08:00
George Hotz
a8de233e12 only div, no reciprocal (#601) 
* only div, no reciprocal

* remove reciprocal

* fix pad shuffling
 2023-02-25 09:35:03 -08:00
Sohaib
d581a99d90 onnx: lrn (#602) 
Co-authored-by: Sohaib Errabii <errabii.sohaib@gmail.com>
 2023-02-25 09:24:53 -08:00
voidz
94bec40110 moved extras/jit.py -> tinygrad/jit.py (#599) 
* moved extras/jit.py to tinygrad/jit.py

* fixed indent

* removed tinygrad.helpers.DEBUG from jit.py
 2023-02-25 08:32:33 -08:00
Benedikt Mandelkow
7348e9a6c6 add restrict qualifier to inputs in c backend (#593) 
* add restrict qualifier for clang backend convolution inputs/ outputs
see https://godbolt.org/z/Tb9jMxWfx for generated assembly

* enable more checks

* inline fmax to motivate the compiler to inline some more

* fix if else binding power
 2023-02-25 08:32:21 -08:00
George Hotz
2c5e13a513 Reluless (#600) 
* replace relu for maximum

* fix for other backend

* clean up RELU and GT0

* tests for maximum

* had to clean that up

* why reverse a maximum?
 2023-02-25 01:21:16 -08:00
George Hotz
176ad29974 retain support for old onnx 2023-02-24 22:29:54 -08:00
George Hotz
f3386c7f09 improve symbolic, hlop conv output is simple now 2023-02-24 22:20:40 -08:00
George Hotz
9b5b17db54 shapetracker : always 0 out the strides if the shape is 1 2023-02-24 21:55:38 -08:00
George Hotz
f8f026e8bb oversized expand for HLOP convs 2023-02-24 21:48:47 -08:00
George Hotz
2edfe64512 improve shapetracker tests 2023-02-24 21:07:53 -08:00
Marcello Fuschi
6d97d62ab3 Add PyCharm's .idea to .gitignore (#597) 2023-02-24 20:14:38 -08:00
George Hotz
da5643d024 rest of tests shouid be made to pass 2023-02-24 12:52:23 -08:00
George Hotz
85452fbaf3 onnx 58/109/208 2023-02-24 12:19:05 -08:00
George Hotz
e8a153e4e9 onnx : add a whole bunch of ops 2023-02-24 12:00:03 -08:00
George Hotz
f2486a7248 more onnx ops 2023-02-24 10:55:58 -08:00
George Hotz
4d0a3dd653 openpilot expand is bugged 2023-02-24 10:25:59 -08:00
George Hotz
2e56a4793e rename log_softmax, support dim, fix onnx Softmax 2023-02-24 10:11:24 -08:00
George Hotz
5cdfeffe2c fix shape test 2023-02-24 09:36:32 -08:00
George Hotz
3becefa218 fix onnx tests 2023-02-24 09:27:18 -08:00
George Hotz
e263c0c628 onnx : another model test is passing 2023-02-24 09:22:58 -08:00
George Hotz
d3feea302d much cleaner way to write onnx ops 2023-02-24 08:46:28 -08:00
George Hotz
d3029c91c5 no rng for op test 2023-02-24 00:23:20 -08:00
George Hotz
f6d946853c more bugfixes 2023-02-24 00:21:29 -08:00
George Hotz
b1b2d8f440 onnx : some op tests working 2023-02-23 23:58:13 -08:00
George Hotz
b287b1d529 fix yolov8 to get to ConvTranspose 2023-02-23 22:46:48 -08:00
George Hotz
2d59b25ead onnx backend test : enable only the model tests 2023-02-23 22:36:26 -08:00
George Hotz
5b10dfcab8 onnx tests : 22/175/208 2023-02-23 22:00:16 -08:00