George Hotz
ea3fa07c2a
bump tinygrad to 0.5, move reshape logic from mlops
2023-02-28 18:07:03 -08:00
George Hotz
e9e71fbfc4
remove mlop ( #619 )
...
* remove mlop
* lil simpler
2023-02-28 17:58:24 -08:00
George Hotz
6b423b675d
fix mulacc when both strides are 0
2023-02-28 17:27:52 -08:00
George Hotz
4c4d88aad4
fix the last bug, and make HLOP the default
2023-02-28 17:04:28 -08:00
George Hotz
fde6c2d62b
fix image grouping
2023-02-28 16:50:46 -08:00
George Hotz
17c55f051d
fix test symbolic
2023-02-28 16:37:08 -08:00
George Hotz
28f52f7c24
improve symbolic
2023-02-28 16:21:58 -08:00
George Hotz
1702a5779f
remove hacks from can_merge
2023-02-28 15:30:20 -08:00
George Hotz
e21df1701b
distribute + refactor merge_views
2023-02-28 14:57:56 -08:00
George Hotz
7e6edfbc64
unbreak onnx conv padding
2023-02-28 13:55:03 -08:00
George Hotz
7d556ca7e0
avg/max pool work in N-D
2023-02-28 13:38:27 -08:00
George Hotz
dcb50a3a9f
better hlop image conv
2023-02-28 13:06:58 -08:00
George Hotz
9d539b8ebb
more intuitive output shape from _pool
2023-02-28 11:41:48 -08:00
George Hotz
d722ffbd04
_pool2d -> _pool
2023-02-28 11:35:19 -08:00
George Hotz
3c8da6bd03
add typing
2023-02-28 10:54:46 -08:00
George Hotz
922f96e527
DeviceBuffer : shape can be correct type now
2023-02-28 10:08:55 -08:00
George Hotz
a8bbcccc16
debug print shapetrackers
2023-02-28 08:11:40 -08:00
George Hotz
cfa5a12f13
simplify in shapetracker
2023-02-28 00:35:26 -08:00
George Hotz
8478a61cdb
simplify in shapetracker
2023-02-28 00:35:26 -08:00
George Hotz
262f81d795
applegpu everywhere
2023-02-27 22:54:59 -08:00
George Hotz
d584bae5c0
fine, openpilot can have 197 kernels
2023-02-27 11:48:36 -08:00
George Hotz
7b999add1d
all onnx model tests pass
2023-02-27 11:22:45 -08:00
George Hotz
652d48ccec
onnx : openpilot expand issue was fixed yesterday. remove hack
2023-02-27 11:04:42 -08:00
George Hotz
9d6b63f043
add ConstantOfShape
2023-02-27 10:57:50 -08:00
George Hotz
082134952b
CastLike works with one type hack
2023-02-27 10:51:26 -08:00
Jacky Lee
1ffe8d68d5
Add more onnx ops ( #615 )
...
* Add Celu
* Add thresholded relu
* Add softsign
2023-02-27 10:43:41 -08:00
George Hotz
643e8b0388
fix tests, test bn evaluate too
2023-02-27 10:39:47 -08:00
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