George Hotz
c003be7309
Revert "track size in shapetracker" ( #3043 )
...
* Revert "track size in shapetracker (#3026 )"
This reverts commit a8ba1ac08f
2024-01-08 13:13:39 -08:00
George Hotz
a8ba1ac08f
track size in shapetracker ( #3026 )
...
* track size in shapetracker
* shapetracker adapter
* size is an int
* create Buffer with st.size
* only compare the views for the jit
* fix webgpu
2024-01-05 20:15:53 -08:00
George Hotz
1765849937
new lazy, benchmark ( #2878 )
...
* lazy rewrite, try 2
* min fix tests
* pass contig test
* put broken pads back
* move that to realize
* no contig child fixes array packing
* so wrong
* now that's correct
* base children
* fix bind issues
* disable to_image_idx
* fix tests
* that failure shouldn't break other tests
* more fixes
* fix torch
* skip failing tests in CI
* 1e-7
* half is broken
* 1e-6 margin of error
2023-12-20 14:33:21 -08:00
chenyu
172a88e719
skip slow test_indexing on METAL ( #2852 )
...
LLVM still runs and is a lot faster, would be curious to know why.
also reworded some error messages and remove regex check
2023-12-19 12:00:54 -05:00
geohotstan
fec8e9060c
Add simple fancy indexing exceptions ( #2706 )
...
* fancy indexing raise error
* updated error message
* improved error check
* oops
* fixed onnx
* oops typo
* merge
* add full_flatten
* try
* merged and updated some tests
* more cleaning
* done
* temp fix onnx
* try
* add todo in onnx_test
* reword
* gah
2023-12-19 11:23:51 -05:00
chenyu
73cadfbb3c
Remove pytest markers ( #2831 )
...
* remove pytest marker
* fix some, skip some
* tweak
* fix
* skip slow
* skip more
2023-12-18 18:53:28 -05:00
George Hotz
6d6eb9302d
ruff checks the max line length is 150 ( #2734 )
...
* ruff checks the max line length is 150
* fix tensor.py
* a lot more
* done
2023-12-12 17:34:47 -08:00
geohotstan
67ff2b2b18
Formatted test_indexing ( #2688 )
...
* added tensor.clone() for more correct cloning behavior
* some work and randint issue
* formatted
* final cleanups
* oops, bug fix
2023-12-09 11:38:36 -05:00
geohotstan
d02ff21f1a
enable test_index and test_advancedindex ( #2648 )
...
* enable test_index and test_advancedindex with pretty diff
* removed contig
* created set_ helper function
* comment change
* del empty line
---------
Co-authored-by: chenyu <chenyu@fastmail.com >
2023-12-07 19:44:39 -05:00
George Hotz
232ed2af3f
more test cleanups ( #2631 )
...
* more test cleanups
* move test example back
2023-12-05 16:17:57 -08:00
geohotstan
fc00da538d
helper functions for test_indexing.py ( #2615 )
...
* add some helpers
* I think it should all work..
* fixed get_set_tensor
* done
* del import
* bye bye typing
* style
* remove empty lines lol
* deleted dtype arg
* del trailing space
2023-12-05 02:00:41 -05:00
chenyu
5db0cdfbd3
support list of ints (or other Tensorable) in tensor indices ( #2520 )
...
* support list of ints (or other Tensorable) in tensor indices
* enable some index test cases
2023-11-30 12:46:33 -05:00
chenyu
bd941a0df1
first version of test_indexing ( #2515 )
...
* first version of test_indexing
* move to test/imported
2023-11-30 00:03:59 -05:00
