3 people 54c15d74a4 python float8 support (#11960) ...

* basic support

* alu

* nan in exec_alu

* rand_for_dtype

* inf + 0.0

* finfo

* revert rand_for_dtype

* clean

* truncate fp8s inf

* spec ok

* float_to_fp8 nan/inf

* least_upper_dtype

* clean up

---------

Co-authored-by: b1tg <[email protected]>

2025-09-18 09:17:09 -04:00

..

__init__.py

more test cleanups (#2631)

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

test_allreduce.py

minimal mstack pr to fix allreduce (#10649)

2025-06-05 15:14:53 -07:00

test_attention.py

rope cache optim for jit prune in llm.py (#11678)

2025-08-28 08:31:29 -07:00

test_block_reorder.py

Move Ops.SPECIAL arg to src (#11918)

2025-09-04 09:31:44 +02:00

test_conv.py

move high level stuff to unit tests [pr] (#10708)

2025-06-08 14:05:56 -07:00

test_device.py

bump test_module_runs to 30 seconds (#12174)

2025-09-14 16:48:40 -04:00

test_disk_cache.py

helpers.diskcache_clear (#4436)

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

test_disk_tensor.py

remove llvm_bf16_cast (#12075)

2025-09-08 20:51:15 +03:00

test_dtype_spec.py

python float8 support (#11960)

2025-09-18 09:17:09 -04:00

test_dtype.py

nbytes should raise an exception when size is unlimited (#11928)

2025-09-03 07:06:20 -07:00

test_elf.py

CLANG -> CPU (#9189)

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

test_gguf.py

fix test_dequantization_mxfp4 (#12123)

2025-09-11 14:22:06 -04:00

test_gradient.py

rename lazydata to uop (#10698)

2025-06-08 08:42:22 -07:00

test_graph_rewrite.py

add dtypes.index (#12015)

2025-09-06 06:03:44 +02:00

test_hashing.py

move device tests to test/device + test cleanups (#11735)

2025-08-19 16:02:20 -07:00

test_helpers.py

raise RuntimeError in merge_dicts instead of assert [pr] (#11965)

2025-09-02 17:18:44 -04:00

test_indexing.py

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

test_kernelize.py

move view left to the outer graph prereqs + testing (#10725)

2025-06-09 20:43:25 -07:00

test_linalg.py

move device tests to test/device + test cleanups (#11735)

2025-08-19 16:02:20 -07:00

test_linearizer_rewrite.py

early reduce simplify (#12046)

2025-09-10 21:02:46 +08:00

test_llm_tokenizer.py

delete the arange dim mismatch restriction (#11568)

2025-08-07 13:46:17 -07:00

test_masked_st.py

move high level tests to unit (#10760)

2025-06-10 12:55:44 -04:00

test_microbenchmarks.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_mnist_dataset.py

move high level tests to unit (#10760)

2025-06-10 12:55:44 -04:00

test_pattern_matcher.py

if upat returns self, it's none (#10898)

2025-06-20 12:11:19 -07:00

test_qcom.py

BufferSpec and ProgramSpec [pr] (#7814)

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

test_rearrange_einops.py

move high level tests to unit (#10760)

2025-06-10 12:55:44 -04:00

test_rewrite_bottom_up_gate.py

Move Ops.SPECIAL arg to src (#11918)

2025-09-04 09:31:44 +02:00

test_rewrite_map.py

add dtypes.index (#12015)

2025-09-06 06:03:44 +02:00

test_rewrite_not_ready.py

skip_0 in graph rewrite [pr] (#11627)

2025-08-11 18:29:04 -07:00

test_rewrite_tracked_childen.py

small changes from rangeify (#11682)

2025-08-15 08:45:52 -07:00

test_shapetracker_math.py

[bounty] Remove using reshape to register symbolic shape (#11771)

2025-08-28 12:30:49 -04:00

test_shapetracker.py

Invalid idx (#12067)

2025-09-12 01:42:02 +02:00

test_shm_tensor.py

line reduction [pr] (#7296)

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

test_simple_schedule.py

simple schedule test [pr] (#10853)

2025-06-17 16:19:27 -07:00

test_simplify_valid_idx.py

Invalid idx (#12067)

2025-09-12 01:42:02 +02:00

test_symbolic_failures.py

[bounty] Z3 symbolic fuzzer [pr] (#10514)

2025-05-28 16:28:37 -04:00

test_symbolic_shapetracker.py

Upgrade some divmod folding for symbolic divs (#12216)

2025-09-17 03:00:50 +02:00

test_tar.py

accept filename decorator [pr] (#8049)

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

test_tensor_data.py

test_tesnor_data to unit/ (#12013)

2025-09-04 19:58:27 -04:00

test_tensor_io.py

tar_extract with Tensors (#7853)

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

test_tensor_uop_representation.py

remove tests that pre date the uop spec (#12168)

2025-09-14 18:47:42 +03:00

test_tqdm.py

Change rate to int in test_tqdm (#10848)

2025-06-18 08:40:43 -04:00

test_transcendental_helpers.py

TRANSCENDENTAL_SUPPORTED_DTYPES->TRANSCENDENTAL_DTYPES (#11752)

2025-08-20 10:29:36 -04:00

test_uop_resolve.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_uop_spec.py

support for tuple ranges (#11890)

2025-08-28 07:02:31 -07:00

test_uop_symbolic.py

Upgrade some divmod folding for symbolic divs (#12216)

2025-09-17 03:00:50 +02:00

test_uop_vmin_vmax.py

Invalid idx (#12067)

2025-09-12 01:42:02 +02:00

test_upat_compile.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_view.py

Fix division by zero and mask bug in add views (#11088)

2025-07-07 10:05:47 -07:00

test_viz.py

remove PROFILE=1 option, it's just VIZ=1 [pr] (#12176)

2025-09-15 12:51:50 +03:00

test_winograd.py

delete more tests (#12043)

2025-09-05 15:31:30 -07:00