ttomsa 76bd4c7d5f advanced setitem (#6262) ...

* advanced setitem draft

* add setitem tests

* fix for tests

* small change

* handle repeated indices with test

* fix v broadcasting to mask

* clean up a bit

* open more tests

* clean up, fixes issue with scalar tensor index

* fix

* fix index_put_ and linter

* add type annotation

* done

* remove non contiguous hack

* woops linter

* name fix

* add back type notation

* more type notation

* final

* linter

* check lazydata not shared

* no numpy

* no numpy

* rename

* index benchmark

* linter

* no cloning time

* rm benchmark

* new function

* rm contiguous and cast early

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
Co-authored-by: chenyu <chenyu@fastmail.com>

2024-09-24 22:14:59 -04:00

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_indexing.py

advanced setitem (#6262)

2024-09-24 22:14:59 -04:00