George Hotz 53f0b2aad7 fix infinite loop in flash attention (#9827) ...

* fix infinite loop in flash attention

* get_contraction_with_reduce

* skip that test

* SINGLE_KERNEL_SOFTMAX + fix multi

* default IGNORE_OOB

* print change

2025-04-10 20:06:44 +08:00

..

__init__.py

more test cleanups (#2631)

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

test_allreduce.py

allow double copy [pr] (#8559)

2025-01-10 18:21:01 -05:00

test_attention.py

move cast to before softmax in attention (#9213)

2025-02-24 17:24:59 -05:00

test_device.py

fix device canonicalize for :0 in middle [pr] (#8193)

2024-12-12 16:32:36 -05:00

test_disk_cache.py

helpers.diskcache_clear (#4436)

2024-05-05 14:19:01 -04:00

test_disk_tensor.py

hotfix: load >2 GiB from disk on macOS (#9361)

2025-03-07 14:51:58 +08:00

test_elf.py

CLANG -> CPU (#9189)

2025-02-20 18:03:09 -05:00

test_gguf.py

add a missing unittest.main() [pr] (#8274)

2024-12-16 14:28:10 -05:00

test_gradient.py

allow VIEW(BUFFER) in Tensor UOps [pr] (#9210)

2025-02-24 13:06:15 +01:00

test_graph_rewrite.py

Acc first (#9232)

2025-02-25 22:10:15 -05:00

test_helpers.py

fix infinite loop in flash attention (#9827)

2025-04-10 20:06:44 +08:00

test_linearizer_rewrite.py

use Ops.REDUCE (#9721)

2025-04-04 10:14:34 +08:00

test_pattern_matcher.py

fix minimum length in pattern matcher (#9736)

2025-04-04 14:57:01 +08:00

test_qcom.py

BufferSpec and ProgramSpec [pr] (#7814)

2024-11-21 12:18:05 +08:00

test_rewrite_map.py

multi stage graph_rewrite_map (#9803)

2025-04-09 15:59:45 +08:00

test_shapetracker_math.py

[bounty] Stride is flip (#8792)

2025-01-31 11:34:10 +09:00

test_shapetracker.py

rewriter -> devectorizer [pr] (#9147)

2025-02-18 12:42:08 +08:00

test_shm_tensor.py

line reduction [pr] (#7296)

2024-10-25 17:05:09 +07:00

test_simplify_valid_idx.py

example test use z3 to verify valid simplification (#9684)

2025-04-02 01:05:52 -04:00

test_tar.py

accept filename decorator [pr] (#8049)

2024-12-05 11:40:59 +08:00

test_tensor_io.py

tar_extract with Tensors (#7853)

2024-12-04 17:03:19 +08:00

test_tensor_uop_representation.py

is_realized only if buffer is allocated (#9253)

2025-02-26 08:58:08 +01:00

test_tqdm.py

add rounding to tqdm unit scale (#9507)

2025-03-19 12:08:46 -04:00

test_transcendental_helpers.py

Vectorize Transcendental Regression Tests (#9753)

2025-04-08 01:27:39 +08:00

test_uop_resolve.py

rename MathTraits to maximum [pr] (#7356)

2024-10-29 16:43:04 +08:00

test_uop_symbolic.py

symbolic fold double where (#9436)

2025-04-05 05:12:17 -04:00

test_uop_vmin_vmax.py

improve vmin/vmax for IDIV (#9678)

2025-04-01 23:16:01 -04:00

test_verify_ast.py

CLANG -> CPU (#9189)

2025-02-20 18:03:09 -05:00

test_view.py

Small bug in _reshape_mask (#8538)

2025-01-08 22:11:24 -05:00

test_viz.py

add err handling tests to viz + cleanups (#9825)

2025-04-10 14:05:05 +08:00