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
13,471 commits 280 branches 15 tags 325 MiB
Commit graph

13,471 commits

This branch
This branch
All branches
Author SHA1 Message Date
chenyu
42f6cf3a90
 		tighter test_real_world mem and kernel count bounds (#13573) 
also check if actual usage is within 20% of set limit, the old limits are too big to be useful
 2025-12-04 13:35:39 -05:00
chenyu
89f9e1dcd5
 		add SGD to beautiful_mnist (#13571) 2025-12-04 12:17:29 -05:00
qazal
512a8f3dd4
 		viz: start global memory PMC tests (#13569) 2025-12-05 00:40:27 +08:00
chenyu
7df56d3b99
 		Optimizer.device is a property (#13568) 2025-12-04 09:25:15 -05:00
nimlgen
db99a61fad
 		qcom: support cpu mappings (#13565) 
* test

* qcom: support cpu mappings

* clean

* msg
 2025-12-04 14:50:46 +03:00
George Hotz
bd6a068ef7
 		move track_rewrites to outer schedule cache (#13556) 
Co-authored-by: qazal <qazal.software@gmail.com>
 2025-12-04 19:13:45 +08:00
qazal
3eae146139
 		faster process replay [pr] (#13564) 2025-12-04 18:52:07 +08:00
Rory Clear
6eab756578
 		fix and test loading num_batches_tracked (#13538) 
* fix and test loading num_batches_tracked

* add failing reverse case

* try reshape state dict if mismatch

* reshape for () and (1,)

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-12-04 01:22:49 -08:00
nimlgen
877a7fdd61
 		jit: support encdec (#13563) 
* jit: support encdec

* fix
 2025-12-04 11:58:34 +03:00
Douglas Nyberg
a8a62bc08e
 		add max/min reduction support to ScatterND (#13562) 2025-12-04 00:53:47 -08:00
ayanhan
edf929ec9d
 		fix: add __delitem__ to Tensor with proper TypeError (#13561) 2025-12-04 00:53:08 -08:00
Douglas Nyberg
9411ecedc4
 		fix CUDA half-precision trunc() type mismatch (#13559) 2025-12-03 21:53:16 -05:00
ayanhan
92b40290c7
 		fix: add test_sum_int and remove outdated TODO in test_custom_kernel (#13560) 2025-12-03 21:51:58 -05:00
Christopher Milan
0a54434b15
 		mitigate ctypes c_bool bitfield bug (#13558) 
* mitigate ctypes c_bool bitfield bug

* don't delete old test
 2025-12-03 20:46:04 -05:00
George Hotz
96d16675fe update examples/gradaccum_mnist.py to use the JIT 2025-12-03 16:11:42 -08:00
George Hotz
24ca8eeaa7
 		small fixups from schedule_cache (#13557) 2025-12-03 15:41:16 -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
a4c4e48385
 		add LUNIQUE op (#13554) 2025-12-03 14:34:34 -08:00
George Hotz
a909cd4581
 		faster HEVC decode (#13552) 
* faster HEVC decode

* bind to variables

* cleanups

* more cleanups
 2025-12-03 11:33:05 -08:00
chenyu
22777a89ea
 		minor test_uop_symbolic updates (#13551) 2025-12-03 13:17:44 -05:00
chenyu
a205f98ef4
 		tighter bound for MOD (#13550) 2025-12-03 11:24:29 -05:00
nimlgen
fcdb01abe7
 		hip: fix ioctl (#13548) 2025-12-03 16:40:43 +03:00
qazal
aab7535805
 		viz: format buffer size unit (#13547) 2025-12-03 21:35:49 +08:00
nimlgen
daea1161cc
 		nv: nvdec for blackwell (#13546) 2025-12-03 16:30:22 +03:00
nimlgen
549f3287a8
 		fix caching for fetch (#13544) 2025-12-03 14:34:14 +03:00
qazal
8390de39e6
 		amd: static flag check for sqtt/pmc (#13545) 2025-12-03 18:36:15 +08:00
George Hotz
ddf3f2d0c4
 		rdna3 asm + zip_extract (#13499) 
* rdna3 asm + zip_extract

* include sqtt

* fix end parsing

* disassembler working

* parsing fields

* instruction

* op

* more parsing
 2025-12-02 22:56:01 -08:00
George Hotz
6bd355fa26
 		add needs_second_gpu decorator (#13543) 
* add needs_second_gpu decorator

* more skips

* two more fixes
 2025-12-02 19:08:23 -08:00
wozeparrot
0d55aec605
 		fix after end (#13542) 2025-12-02 18:42:58 -08:00
chenyu
8902781dc1
 		enable more benchmarks (#13540) 
* enable more benchmarks

* disable some

* adjust ASSERT_MIN_STEP_TIME

* mac NOCLANG=1
 2025-12-02 20:31:14 -05:00
George Hotz
055d5aeb7f add external_test_process_count 2025-12-02 17:26:30 -08:00
chenyu
e8879f7e31
 		match torch clamp backward (#13533) 
* match torch clamp backward

* fix PYTHON
 2025-12-02 17:58:32 -05:00
qazal
7622be761f
 		add new remu instructions from #13533 (#13539) 2025-12-03 06:29:20 +08:00
wozeparrot
18640f57b2
 		feat: configurable timeout (#13537) 2025-12-02 13:35:35 -08:00
chenyu
21aac568fd
 		limit lift x*y out of reduce to int [pr] (#13535) 2025-12-02 16:11:45 -05:00
Roelof van Dijk
c158e3c988
 		add cifar gated uop_given_valid regression test (#13536) 2025-12-02 16:02:47 -05:00
Roelof van Dijk
e329baffa7
 		fix cifar while keeping openpilot fused (#13528) 
* this works

* test now passes
 2025-12-02 12:05:56 -08:00
nimlgen
0874ba8cc8
 		test_hevc: do not download the whole file (#13531) 
* test_hevc: do not download the whole file

* fix
 2025-12-02 21:31:28 +03: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
21184ae6b1
 		bump cache to 14 (#13530) 2025-12-02 08:02:19 -08:00
George Hotz
037edc151c
 		late gate for ALLOW_TF32 (#13527) 
* remove ALLOW_TF32

* the right place to put that gate
 2025-12-02 07:51:58 -08:00
Douglas Nyberg
6a7c58abf1
 		fix(onnx): unwrap list/tuple value in Pad op (#13500) 
* fix(onnx): unwrap list/tuple value in Pad op

* add regression test for Pad list value

* remove trailing whitespace

* use _resolve_const for Pad constant_value
 2025-12-02 07:47:20 -08:00
qazal
c65aa93081
 		refactor sqtt loader to enable PMC=1 SQTT=0 (#13526) 2025-12-02 22:50:38 +08:00
chenyu
60f7c6cce6
 		simpler drop_and_clauses [pr] (#13525) 2025-12-02 09:12:21 -05:00
nimlgen
77a76d1b13
 		device: respect compiler ContextVars (#13523) 
* device: envvars for cc

* fix

* fix

* x

* um

* fix

* remote

* em

* cleanup

* typing

* fix

* debug

* lvp?

* ugh

* singl

* rm

* lol

* fix

* ?

* this?

* why?

* rev

* mod test

* l
 2025-12-02 14:42:04 +03:00
wozeparrot
1b7dbfb37f
 		tk: named kernels + per kernel range id (#13522) 2025-12-01 22:51:04 -08:00
wozeparrot
8713ae6de9
 		fix: dead sdv2 download link (#13521) 2025-12-01 22:50:53 -08:00
George Hotz
44104b0b7f
 		mnist with grad acc + Adam on CPU (#13520) 
* mnist with grad acc + Adam on CPU

* still broken, but closer

* works w/o jit

* this works without the jit
 2025-12-01 18:27:32 -08:00
George Hotz
7307120311
 		shard to one device is to (#13519) 
* shard to one device is to

* fst
 2025-12-01 16:29:53 -08:00
chenyu
0b92fd30f5
 		simpler simplify_valid [pr] (#13514) 
dedup instead of getting a True clause which is removed later
 2025-12-01 17:36:33 -05:00