George Hotz
b132de677d
tinygrad.nn ( #367 )
...
* tinygrad.nn
* flake8
* working on pylint
* more pylint
* more pylint
* pylint passes
* networkx
* mypy can't infer that type
* junk
2022-08-18 07:41:00 -07:00
George Hotz
3c4565fa21
SLICE -> PAD,SHRINK
2022-07-17 11:33:59 -07:00
George Hotz
cca089b11d
Revert "more expand -> repeat"
...
This reverts commit 2e7b1630a8 .
2022-07-17 08:41:48 -07:00
George Hotz
2e7b1630a8
more expand -> repeat
2022-07-17 08:40:49 -07:00
George Hotz
cde137d163
simple shapetracker tests
2022-07-02 16:02:15 -07:00
George Hotz
c833886bf5
improved shapetracker
2022-06-21 19:17:25 -07:00
George Hotz
a7131b6a46
Non contig ( #339 )
...
* contiguous_view
* non contig reduce too
* conv fast
* maybe faster valid
* improve test_onnx
* improve params
* elementwise_op
* draw non contig
* improve contiguous
2022-06-19 22:40:48 -07:00
George Hotz
fb72ea3fbd
gpu uses shapetracker (fix tests) ( #335 )
...
* shapetracker
* movement_op
* hmm, that's why repr failed
2022-06-19 17:32:07 -07:00
George Hotz
6b652dafb2
touchups
2022-06-19 16:57:14 -07:00
George Hotz
e364849b3b
stuff from lazy
2022-06-19 09:57:16 -07:00
George Hotz
85fe25e27b
add stride support to shapetracker
2022-06-15 17:48:41 -07:00
George Hotz
6261a0639b
ShapeTracker ( #328 )
...
* start shapetracker
* that late reshape is crushing our hopes
* simple failure
* DumbShapeTracker passes tests
* improve st tests
* stacked view tracker works
* flip works
* tests pass
* shapetracker works
* use ShapeTracker in ops_gpu
* a couple lines
* fix 0 shape
* less lines
* use shapetracker for new_shape in ops.py
* simpler still
* padding with a ZeroView
* gamed it a little
2022-06-14 16:08:22 -07:00