George Hotz b4bf6a7dea switch backward to use gradient [pr] (#8235) ...

* switch backward to use gradient [pr]

* set device correctly, dedup

* why does that fail?

* add noop cast

* simple backward

* fix beautiful_mnist

* touchups

* set in compute_gradient

* uop_count

* uop_count was wrong

* collections

* no note

* skip that test

* update sched kernel counts

* train mnist is 65

* fix metadata and gc

* fixes

* materialize_grads

* no pathlib stuff

* add contiguous_backward, fix bugs

* add some realize

* fix multi

2025-01-26 09:12:16 +09: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_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

don't use view in copy [pr] (#8704)

2025-01-21 09:57:47 -08:00

test_elf.py

run unit tests separate from others (and only once) (#6020)

2024-08-10 11:17:56 -07:00

test_gguf.py

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

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

test_gradient.py

switch backward to use gradient [pr] (#8235)

2025-01-26 09:12:16 +09:00

test_graph_rewrite.py

Move define_acc down an unrolled add chain (#8404)

2025-01-20 14:56:27 -05:00

test_helpers.py

simpler helpers.Context [pr] (#8433)

2024-12-27 17:27:14 -05:00

test_pattern_matcher.py

Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)

2024-11-16 20:56:56 +08:00

test_qcom.py

BufferSpec and ProgramSpec [pr] (#7814)

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

test_rewrite_map.py

hotfix: remove duplicate TestTensorMutates [pr] (#8619)

2025-01-14 16:03:17 -05:00

test_shapetracker_math.py

more typing cleanups [pr] (#8376)

2024-12-22 05:21:03 -05:00

test_shapetracker.py

rename uopgraph to rewriter [pr] (#8682)

2025-01-19 17:03:12 -08:00

test_shm_tensor.py

line reduction [pr] (#7296)

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

test_simplify_valid_idx.py

rename uopgraph to rewriter [pr] (#8682)

2025-01-19 17:03:12 -08: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

don't use view in copy [pr] (#8704)

2025-01-21 09:57:47 -08:00

test_tqdm.py

Context manager support for tqdm (#7770)

2024-11-18 14:12:03 -05:00

test_transcendental_helpers.py

update payne_hanek_reduction [pr] (#7455)

2024-10-31 18:41:22 -04:00

test_uop_resolve.py

rename MathTraits to maximum [pr] (#7356)

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

test_uop_symbolic.py

rename uopgraph to rewriter [pr] (#8682)

2025-01-19 17:03:12 -08:00

test_uop_vmin_vmax.py

type annotation of resolve [pr] (#8467)

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

test_verify_ast.py

use UOp.st for kernel reduce axes (#8499)

2025-01-13 06:24:11 -05:00

test_view.py

Small bug in _reshape_mask (#8538)

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

test_viz.py

viz cleanups 2 [pr] (#8748)

2025-01-25 19:41:57 +02:00