Francis Lam
162dfb07d9
fuzz_linearizer: fix uops and add to test.yml ( #3588 )
2024-03-02 15:03:42 -08:00
Francis Lam
11da65bccd
test/external/fuzz_linearizer: add a FUZZ_MAX_SIZE option ( #3455 )
...
* test/external/fuzz_linearizer: add a FUZZ_MAX_SIZE option
this allows us to limit the size of the kernel and reduce running
times by avoiding ones that take a long time
* fix spacing and re-order to put parameters together
2024-02-27 07:34:59 -05:00
xarkes
28a8b72024
Remove Interpreted device & remaining CPU/TORCH ref ( #3423 )
...
* Remove Interpreted device & remaining CPU/TORCH ref
* Oops
* supports_device was useful
* Fix doc wording
---------
Co-authored-by: chenyu <chenyu@fastmail.com >
2024-02-16 00:30:21 -05:00
George Hotz
41efaa848c
move graph.py and jit.py into features ( #3376 )
...
* move graph.py into features
* move jit into features
* fix quickstart
2024-02-12 17:34:34 +01:00
Francis Lam
2266152b28
linearizer: added FUZZ_BEAM to fuzz_linearizer and additional tests ( #3340 )
...
Fixed test_tensor_core_opts to test all the TCs.
Added commented out failing tests in test_color_shapes_with_local.
2024-02-08 16:12:58 +01:00
nimlgen
f87ecbb0f3
fuzzer validates outputs + (partially) oob accesses ( #3178 )
...
* fuzzer validates outputs + (partially) oob accesses
* +random
* oob check only for compiled
* type cmp fixes
* fix zeroing
* no prints
* add seed
2024-01-19 13:34:51 -05:00
chenyu
1b508e0f71
fix fuzz_linearizer toCPU to as_buffer ( #3158 )
2024-01-17 13:18:46 -05:00
chenyu
58d3d5030b
vars_from_ast -> LazyOp.vars ( #2965 )
2024-01-01 18:12:38 -05:00
George Hotz
56f44bd10e
move the compiler cache to be global ( #2957 )
...
* move the compiler cache to be global
* remove non robust test
* remove dead code
2024-01-01 10:59:56 -08:00
George Hotz
00d9eda961
FROM -> COPY, move vars_from_ast ( #2675 )
2023-12-07 16:32:30 -08:00
chenyu
51af99367f
fix fuzz_linearizer using new device Buffer ( #2674 )
2023-12-07 19:21:47 -05:00
Christopher Mauri Milan
7f01dd04f0
Apply ruff linting rules to tests ( #2473 )
...
* everything except F821
* enable F821 with noqa
* dumb fix
* fix remaining imports and (former) lambdas
* replace _ with noqa to avoid gc
2023-11-27 21:24:06 -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
George Hotz
8e9cdef61f
clean up the buffers ( #2447 )
...
* clean up the buffers
* remove allocate_output
* functools.lru_cache is methodcache
* add TestShapeTrackerSize
* cache_clear
* no 0 sz buffer, add _ on functions that shouldn't be imported
* fix size
* if -> while
2023-11-26 11:02:29 -08:00
George Hotz
0505c5ea50
remove force_wait, refactor to graph ( #2405 )
...
* remove force_wait
* refactor
* get rid of stupid ASTRunner
* fix del in diskbuffer
* BufferOps.FROM_UNDERLYING
* put offset in the rawbuffer
* fix bugs
* use exec
2023-11-23 12:46:07 -08:00
chenyu
a98511561c
fuzz_linearizer same api for interpreted and compiled ( #2320 )
2023-11-15 17:40:22 -05:00
chenyu
fac8633ba8
explicit opts for test_linearizer_failures ( #2299 )
...
* explicit opts for test_linearizer_failures
* typo
* update the invalid check
2023-11-14 11:52:38 -05:00
George Hotz
85d26ddc36
uops loop removal ( #2262 )
...
* remove the loop
* cleanups
* tests failing still
* global_loop_ctx wasn't needed
* replace_op is cleaner
* minor opt
* cast opt was wrong
* uop_num
* uop num was dumb
* tuplize_uops
* torch tests
* fix test_uops
2023-11-10 15:24:47 -08:00
chenyu
75f6e9ab54
one more fuzz linearizer failed example ( #2260 )
2023-11-10 09:17:37 -05:00
chenyu
10d642e174
fuzz linearizer transformation ( #2188 )
...
* fuzz linearizer transformation
* no standard normal for fp16
* work
* Interpreted start
* CPU and TORCH work
* fix MemBuffer with same idx
* id for failed kernels
* no image and variable for Interpreted
* symbolic shape
* IMAGE only for GPU
* Interpreted almost all good
* cleanup
* fix bufs_from_lin
* zero size
* some failed examples
* just Exception
* just test not pass
2023-11-09 08:03:27 -08:00
