1
0
Fork 0
Commit Graph

134 Commits (deepcrayon)

Author SHA1 Message Date
Jeff Moe 076b5cf983 new logo, mv benchmarks etc to tinyrocs 2024-01-04 10:26:22 -07:00
Jeff Moe d38f30d907 Use parrot logo 2023-12-30 12:05:41 -07:00
Jeff Moe 26b48648de Update tinyrocs linkies 2023-12-30 12:03:51 -07:00
Jeff Moe 1a1daab143 tinyrocs, benchmarks 2023-12-10 12:46:09 -07:00
Jeff Moe 160fe8b15e tingrab pic, favicon 2023-12-06 18:47:03 -07:00
Jeff Moe f81204dea3 docstrings shape view 2023-12-06 18:25:05 -07:00
Jeff Moe e9183beaac docstrings shapetracker, cstyle, symbolic 2023-12-06 18:06:38 -07:00
Jeff Moe 186f84cc09 docstrings for renderer cstyle, llvmir 2023-12-06 17:05:03 -07:00
Jeff Moe 8c8a212b99 docstrings for tinygrad nn state 2023-12-06 15:06:38 -07:00
Jeff Moe a375c4cfa3 docstrings tinygrad nn optim 2023-12-06 14:53:11 -07:00
Jeff Moe 45dbac0a02 docstrings for tinygrad features graph hip, but fails to build 2023-12-06 14:15:09 -07:00
Jeff Moe c51f2a7dbd docstrings tinygrad features graph cuda 2023-12-06 14:05:11 -07:00
Jeff Moe d2b875aeb0 docstrings features image, search 2023-12-06 12:36:02 -07:00
Jeff Moe 7b95cacfe4 docstrings linearizer, a bit foo 2023-12-06 12:24:08 -07:00
Jeff Moe bd47ccaf04 codegen kernel docstrings 2023-12-06 12:03:19 -07:00
Jeff Moe fe8d9753f0 docstrings, requirements webgpu 2023-12-06 11:31:18 -07:00
Jeff Moe 96fb6d334b Add torch docstrings, and to requirements 2023-12-06 11:03:20 -07:00
Jeff Moe e999513df0 Metal docstrings with fail 2023-12-06 10:53:45 -07:00
Jeff Moe f1af595918 docstrings for llvm 2023-12-06 10:44:45 -07:00
Jeff Moe 042dd69139 ops_gpu docstrings 2023-12-06 10:22:47 -07:00
Jeff Moe 421396343e ops_disk doctrings 2023-12-06 10:14:14 -07:00
Jeff Moe 6b70666dd9 cuda doctrings; tinygrad notice 2023-12-06 10:07:16 -07:00
Jeff Moe dcf9c9438d ops_clang, ops_cpu docstrings 2023-12-06 09:46:46 -07:00
Jeff Moe 4ec99c864f just en for now 2023-12-05 16:38:59 -07:00
Jeff Moe 3a5f68059c partial docstring for tinygrad tensor 2023-12-05 16:38:08 -07:00
Jeff Moe d7736d19ea docstrings for tinygrad lazy 2023-12-05 14:53:21 -07:00
Jeff Moe 6dcc07a557 docstrings for tinygrad mlops 2023-12-05 14:15:55 -07:00
Jeff Moe de93d0fe9e docstrings for tinygrad ops 2023-12-05 13:42:27 -07:00
Jeff Moe 1add458c01 docstrings for tinygrad jit 2023-12-05 13:01:19 -07:00
Jeff Moe c239dc246d tinygrad graph docstrings 2023-12-05 12:32:44 -07:00
Jeff Moe 7bde7cc998 reorg a bit, rm .po 2023-12-05 12:06:42 -07:00
Jeff Moe 9836f5ee62 tinygrad realize docstrings 2023-12-05 11:56:26 -07:00
Jeff Moe 9fdc64ccc9 tinygrad helpers docstrings 2023-12-05 11:44:12 -07:00
Jeff Moe ca23531c21 tinygrad device docstrings 2023-12-05 10:45:53 -07:00
Jeff Moe c13563bc66 make lines perfecto length, like RPG III or something 2023-12-04 22:25:14 -07:00
Jeff Moe 20e93f926c doc formatting etc 2023-12-04 22:22:30 -07:00
Jeff Moe 661dcc5ed0 Reformat, uh, everything, with black 2023-12-04 22:01:04 -07:00
Jeff Moe 01503ca90d sphinx, crufty 2023-12-04 22:00:18 -07:00
Jeff Moe 4b032f66b0 Initial translation strings 2023-12-04 20:55:50 -07:00
Jeff Moe ad752fc417 Set up for tinygrab from Parrot stubs 2023-12-04 20:52:35 -07:00
Jeff Moe cf472ad527 Pull in Sphinx build config, etc from Parrot 2023-12-04 20:40:20 -07:00
Amrit Sahu e8d6a6ef2e
view.reshape without symbolic (#2218)
* handle reshape of contiguous subparts with explicit mask

* remove the add/remove ones logic in reshape

* accomodate ones in accumulate logic

* make multiply commutative

* fix linting

* make mypy happy

* add test for commutative mul

* merge dimensions in shape_strides for 1 range masks

* add offsets for merging

* fix linting

* add back explicit 1 reshapes

* fix mypy errors

* fix accumulate by includng state

* include non-zero stride dimension in acc

* small cleanup

* more compact to_shape_strides

* more logical cleanup

* compress more

* compress reshape mask

* adding some comments

* small bug fix

* improve test coverage

* remove explicit add remove ones

* small bug in test

* enable test_reshape_splitting_combining

* small fix

* 10 lines less to_shape_strides

* shorten reshape mask

* some more cleanup

* more cleanup

* introduce some symbols for compactness

* more symbols

* more cleaner

* lessen symbols, it became less readable

* remove merge_views from view.reshape

* change to_shape_strides to _merge_dims

* improve readability

* fix corner case

* cleanup

* better handling of 1 <= Variable('i',1,10) & new_dim = Variable('i',1,10)

* rewrite _reshape_mask for readability

* fix white space

* add comment

* nice shorthands for readability

* add proof in docs

* small nit

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
2023-12-04 12:46:53 -05:00
George Hotz d6b404ac11
No dtype alloc (#2570)
* fix all allocs

* improve docs

* ugh fix fake alloc
2023-12-02 13:29:40 -08:00
George Hotz 5068e99d18
refactor to remove extra kernel params (#2563)
* refactor to have compiled kernel

* bugfixes

* docs/beautiful.py

* revert that

* fix tests
2023-12-02 00:32:25 -08:00
George Hotz 6733425095
lower schedule (#2559)
* lower schedule

* remove RAND, and don't put load in the JIT yet

* better fix for that test
2023-12-01 19:17:46 -08:00
wozeparrot 28183c7438
feat: reword (#2549) 2023-12-01 10:56:18 -08:00
chenyu 7fec966b5e
bye bye NOOP (#2534)
* bye bye NOOP

* SIN

* NEG
2023-11-30 23:10:35 -08:00
George Hotz 2c363b5f0b
new style device (#2530)
* cpu tests pass

* torch works

* works

* metal works

* fix ops_disk

* metal jit works

* fix openpilot

* llvm and clang work

* fix webgpu

* docs are rly broken

* LRU works on metal

* delete comment

* revert name to ._buf. LRU only on Compiled

* changes

* allocator

* allocator, getting closer

* lru alloc

* LRUAllocator

* all pass

* metal

* cuda

* test examples

* linearizer

* test fixes

* fix custom + clean realize

* fix hip

* skip tests

* fix tests

* fix size=0

* fix MOCKHIP

* fix thneed

* copy better

* simple

* old style metal copy

* fix thneed

* np reshape

* give cuda a device
2023-11-30 17:07:16 -08:00
Yingbo Ma d43485ae9e
Fix `graph_uops` (#2457)
* Load networkx when we need to graph uops

* Document GRAPHUOPS

* import nx in `graph_uops`
2023-11-27 18:42:48 -08:00
George Hotz 9e07824542
move device to device.py (#2466)
* move device to device.py

* pylint test --disable R,C,W,E --enable E0611

* fix tests
2023-11-27 11:34:37 -08:00