mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
12,003 commits 280 branches 15 tags 325 MiB
Commit graph

948 commits

Author SHA1 Message Date
nimlgen
230d08ec70
 		test for am recovery and faults handling (#14421) 
* test for am recovery and faults handling

* linter
 2026-01-29 17:11:24 +03:00
George Hotz
88bc5ee212
 		assembly/amd: rename to better names (#14384) 
* assembly/amd: rename to better names

* might help fuzzing segfault

* emu2 -> emu
 2026-01-28 10:00:54 +08:00
George Hotz
984cdc4840
 		add wrapper class for the -0.0 != 0.0 issue (#14339) 
* add wrapper class for the -0.0 != 0.0 issue

* fixes

* spec fix

* missed one
 2026-01-26 16:52:37 +08:00
nimlgen
26220a472e
 		no core_id (#14265) 
* no core_id

* kwargs

* est

* linters

* ugh

* revert this

* deps

* glb

* should work?

* nn

* line

* fx

* ym

* z

* d

* um?

* revert

* this one?

* first half

* um p2

* all?

* um

* cleaner

* um
 2026-01-23 21:30:12 +03:00
chenyu
073c6a81b5
 		raise if Tensor._buffer is called during jit (#14114) 
* raise if Tensor._buffer is called during jit

* cleaner
 2026-01-22 17:30:18 -05:00
chenyu
574d171fa6
 		fix onnx Pad constant_value=None (#14271) 
also removed a dead branch in _resolve_pool_pads
 2026-01-21 11:51:34 -05:00
chenyu
9ea63d7d52
 		failed test case for onnx IF with jit (#14235) 
silently fails now since onnx treats IF cond as a const
 2026-01-19 18:10:05 -05:00
chenyu
5e6a72c33f
 		new Onnx Gather (#14187) 
instead of assuming const indices, check if it showed as a const
 2026-01-16 22:24:07 -05:00
chenyu
ab244c7f81
 		onnx Gather should not assume indices to be const (#14185) 
* onnx Gather should not assume indices to be const

added a failed test case

* just list
 2026-01-16 20:55:00 -05:00
chenyu
2a2c1eacf6
 		disable fast_idiv on metal (#14137) 
there's a metal compiler bug which was the root cause that keccak needs a contigous hack
 2026-01-13 21:40:40 -05:00
chenyu
cad7feec02
 		more onnx ops (#14104) 
HannWindow, HammingWindow, BlackmanWindow, Hardmax, LpNormalization
 2026-01-12 09:11:13 -05:00
chenyu
9973a81356
 		add channels_last to QLinearGlobalAveragePool (#14094) 
and other minor cleanups
 2026-01-10 18:38:19 -05:00
chenyu
83063cc3e4
 		onnx TensorScatter (#14024) 2026-01-05 09:05:22 -05:00
chenyu
9497ec00f2
 		fix onnx attention permute (#14025) 
* fix onnx attention permute

* skip test_attention_4d_fp16_cpu too
 2026-01-05 08:58:50 -05:00
chenyu
7a81a3cb98
 		more passed onnx tests (#14022) 2026-01-05 07:46:27 -05:00
chenyu
aae08b20e0
 		enable passed onnx tests (#14017) 2026-01-04 22:12:50 -05:00
chenyu
f6a78a29e0
 		support einsum trace (#14012) 
* support einsum trace

* test_einsum_scalar_cpu
 2026-01-04 19:27:27 -05:00
qazal
bdb421f13e
 		process_replay: passthrough sink arg for Ops.PROGRAM input (#14000) 2026-01-04 13:09:39 +09:00
chenyu
51398edf9c
 		fix indirect import (#13958) 
also deleted old external tests
 2026-01-01 14:22:45 -05:00
nimlgen
25440f0f72
 		all2all (#13902) 
* all2all

* um

* fix

* x

* um

* simler

* mypy

* fix

* t

* cmnts
 2025-12-31 16:38:32 +03:00
George Hotz
43c6e973d8
 		add optional compiler in Renderer (#13817) 
* add optional compiler in Renderer [pr]

* fix

* late init

* remove precompiled

* cleanup
 2025-12-23 17:58:46 -05:00
nimlgen
90b217896f
 		am: xgmi p2p (#13811) 
* system: use addr space

* am: xgmi

* fix

* ugh
 2025-12-23 20:11:38 +03:00
George Hotz
8dcba2e2cc
 		no full_rewrite [pr] (#13809) 
* no full_rewrite [pr]

* fix

* fix docs
 2025-12-22 23:20:01 -05:00
chenyu
7f1d41c9f9
 		delete files that import ShapeTracker (#13805) 2025-12-22 15:54:18 -05:00
George Hotz
45c459848d
 		remove more stale stuff (#13765) 
* remove more stale stuff

* remove disassemblers/adreno

* stale
 2025-12-19 17:14:56 -04:00
George Hotz
744af193f0
 		remove ScheduleItem and merge it with ExecItem (#13759) 
* remove ExecItem and merge it with ScheduleItem

* less diff

* fix issues

* min diff

* don't change bufs in _lower

* min diff

* update

* revert

* fixes

* diff
 2025-12-19 17:04:24 -04:00
George Hotz
3dbde178c1
 		mark slow tests as slow instead of as CI (#13736) 
* mark slow tests as slow instead of as CI

* CI shouldn't have different behavior

* more skips / CI

* slow
 2025-12-17 10:29:57 -04:00
George Hotz
4b741e893f
 		remove REMOTE=1 (#13722) 
* remove REMOTE=1

* leave ibverbs
 2025-12-16 15:58:10 -04:00
nimlgen
e36385e570
 		am: support xgmi systems (#13659) 
* am: support xgmi systems

* fake_am
 2025-12-12 18:55:45 +03:00
Douglas Nyberg
947c6eefc3
 		add Swish op (#13541) 
* add Swish ONNX operator

* add Swish regression test

* remove trailing whitespace

* upgrade ONNX to 1.20, add excludes for unimplemented ops

* upgrade ONNX to 1.19, add Swish op

* upgrade ONNX to 1.19, TensorFlow to 2.18, add Swish op

* exclude attention_3d and attention_4d_gqa tests

* exclude attention fp16 tests

* exclude all attention tests

* retrigger CI

* retrigger CI - worker crash
 2025-12-08 12:41:18 -05:00
George Hotz
c5bd28e21d
 		start work on schedule cache (#13529) 
* start work on schedule cache

* local unique

* schedule cache works

* schedule cache cleanup

* fix tests

* preserve metadata

* oops, fix cache

* put that there

* fix spec

* always miss

* why is that broken?

* src[0].op

* fix process replay

* delete abstractions2

* reenable the actual schedule cache

* metadata is best effort

* fix JIT in examples/gradaccum_mnist.py

* full jit

* fixed and test is real
 2025-12-04 17:24:49 -08:00
Douglas Nyberg
a8a62bc08e
 		add max/min reduction support to ScatterND (#13562) 2025-12-04 00:53:47 -08:00
Douglas Nyberg
f5abd38132
 		remove tfa dependency: use keras.optimizers.Lamb and tf.raw_ops for LARS (#13555) 2025-12-03 17:48:27 -05:00
George Hotz
055d5aeb7f add external_test_process_count 2025-12-02 17:26:30 -08:00
qazal
366badaa68
 		require renderer argument in get_program, removes device opening in process replay [pr] (#13524) 2025-12-03 02:05:31 +08:00
George Hotz
6a140f74fe
 		split out unique_const and cache const [pr] (#13493) 
* split out unique_const

* add cache to const

* call const in unique_const
 2025-11-29 10:44:28 -08:00
George Hotz
18addc0a1d
 		process replay only get_program (#13475) 2025-11-27 08:18:18 -08:00
George Hotz
a8e005b095
 		enable process replay (non-checking) by default (#13474) 2025-11-27 07:28:44 -08:00
George Hotz
05cd2279d0
 		add cache on reshape (#13466) 
* remove cache on divmod, way less objects

* _apply_reshape

* reshape

* no gc on realize

* wow that cache is fast
 2025-11-26 18:57:40 -08:00
Sieds Lykles
63a931ff76
 		Symbolic divisor fuzzer (#13433) 
* render z3 range better

* working version

* rename

* add to workflow

* factor out variable_names

* smaller expressions

* smaller

* + back
 2025-11-23 20:29:32 +01:00
qazal
9dcd52287a
 		add external_benchmark_pyrender (#13378) 
* add external_benchmark_pyrender

* can ctrlc it

* cpu_profile exists
 2025-11-20 17:38:28 +08:00
George Hotz
8919c994b7
 		Revert "AxisType.PLACEHOLDER in reshape to do less graph_rewrite (#13373)" (#13375) 
This reverts commit ac7559e33d.
 2025-11-19 19:34:30 -08:00
George Hotz
ac7559e33d
 		AxisType.PLACEHOLDER in reshape to do less graph_rewrite (#13373) 
* AxisType.PLACEHOLDER in reshape to do less graph_rewrite

* _apply_movement_op cache
 2025-11-19 19:19:58 -08:00
George Hotz
ab7df42c78
 		bring back fold_divmod_general with bugfix and test [pr] (#13369) 
* Revert "Revert "merge to fold_divmod_general [p] (#13359)""

This reverts commit 05ccc69248.

* Revert "Revert "actually merge to fold_divmod_general [pr] (#13363)""

This reverts commit 90e5752199.

* Revert "Revert "add cache to fold_divmod_general (#13365)""

This reverts commit 8e17bd6791.

* bring back fold_divmod_general with bugfix and test
 2025-11-19 14:51:51 -08:00
George Hotz
8e17bd6791 Revert "add cache to fold_divmod_general (#13365)" 
This reverts commit b5309a5043.
 2025-11-19 14:18:08 -08:00
George Hotz
b5309a5043
 		add cache to fold_divmod_general (#13365) 2025-11-19 13:49:18 -08:00
George Hotz
6fdbd03104
 		more divmod cleanup [p] (#13358) 
* more divmod cleanup [p]

* lil cleanups, faster
 2025-11-19 10:35:15 -08:00
George Hotz
385618d45b
 		skip process replay by default (#13353) 2025-11-19 08:25:34 -08:00
George Hotz
6d3385c284
 		print special ops in postrange (#13318) 
* print special ops in postrange

* fix on OSX
 2025-11-17 14:43:23 -08:00
George Hotz
e5351699bd
 		openpilot warp (#13283) 
* openpilot image warp test

* 0.4 ms on metal, 1 ms on CPU

* new inputs each time

* reshape
 2025-11-14 13:55:32 -08:00