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

5,175 commits

Author SHA1 Message Date
George Hotz
de3e901b71 works but bad 2026-02-24 14:40:39 +08:00
chenyu
0bda5585c7
 		unit test TestTinyFS (#14972) 
these passed before the allocation change
 2026-02-23 16:59:39 -05:00
imaolo
405d37423e
 		call release() in MetalAllocator._free (#14970) 
* add failing test

* call MTLBuffer.release() in MetalAllocator._free()

* Update test_metal.py

---------

Co-authored-by: nimlgen <138685161+nimlgen@users.noreply.github.com>
 2026-02-23 23:33:31 +03:00
chenyu
127136421d
 		enable a few WEBGPU isnan tests that work now (#14967) 
* enable a few WEBGPU isnan tests that work now

* still failed
 2026-02-23 11:06:08 -05:00
ttomsa
0366474089
 		Bool cast to cmpne (#14544) 
* test

* rm in llvmir

* rm in ptx and nir

* hmmmm

* rm in decompositions

* skip tests

* add test

* just this

* rm comment

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2026-02-23 10:31:36 -05:00
George Hotz
b824490e3f
 		allocate generates a call (#14958) 
* allocate generates a call

* symbolic works too

* DEFINE_VAR is param

* replace param later

* apply buffers

* name

* upd

* this was a bug...
 2026-02-23 15:59:20 +08:00
wozeparrot
25565b2410
 		fa: test for mp (#14907) 2026-02-22 21:47:36 -08:00
qazal
d6145736c7
 		sqtt: examples generator changes from inst_discovery (#14961) 
* sqtt examples generator changes from inst_discovery

* rdna4

* rdna3

* cdna

* sad reality for mi300x
 2026-02-23 14:42:48 +09:00
George Hotz
677145b393
 		all consts have shapes (#14959) 
* all consts have shapes

* vconst has shape too

* use normal schedule

* cast ptrdtype

* image

* bitcast issue + hack
 2026-02-23 10:26:50 +08:00
qazal
1538960002
 		viz: smaller view for repeated asm instructions in cfg (#14954) 
* simple test

* todo

* feature
 2026-02-23 10:41:43 +09:00
chenyu
4424757b9a
 		update test_sharded_memory (#14956) 
cleaned up and moved to test/null
 2026-02-22 16:56:08 -05:00
b1tg
f9b7493e7a
 		cleanup fp8 conversion helpers and fp8 edge-case tests (#14953) 
Co-authored-by: b1tg <b1tg@users.noreply.github.com>
Co-authored-by: chenyu <chenyu@fastmail.com>
 2026-02-22 09:16:42 -05:00
qazal
60f90dd97c
 		sqtt: fix jitted program deduping, failing test for graphed kernels (#14951) 
* work

* hcq_profile fix, test with JIT=2 passes

* ci, -n=auto

* rm duplicate test

* less
 2026-02-22 15:22:31 +09:00
chenyu
24e8919438
 		raise explicitly for test_crossunder_assign (#14948) 2026-02-21 21:21:13 -05:00
chenyu
9764e2561c
 		more assign into unrealize silent fail cases (#14944) 2026-02-21 18:12:57 -05:00
nimlgen
6de15dc480
 		mockam usb (#14916) 
* mockam usb

* f

* win

* x

* x
 2026-02-21 23:05:54 +03:00
chenyu
0dbcd764ad
 		a few assign into unrealized failed test case (#14940) 2026-02-21 13:18:45 -05:00
chenyu
0255a64a27
 		update test_jit_init_empty (#14938) 
* update test_jit_init_empty

now it fails silently

* that
 2026-02-21 09:01:50 -05:00
George Hotz
8ef5544e4a
 		realized PYTHON copies (#14934) 
* realized PYTHON copies

* comment that out

* fix that test

* append afters

* contig

* disk copies

* should be 124

* 332
 2026-02-21 20:29:31 +08:00
qazal
8278886cf9
 		test_profiler cleanup, non flaky cpu_profile test (#14932) 
* test_profiler cleanup, non flaky cpu_profile test

* existing device is okay
 2026-02-21 16:58:10 +09:00
qazal
c5029fa460
 		jit case with Tensor.empty input, realized means allocated (#14930) 
* simple failing jit test case with Tensor.empty

* this used to exist in ops.py...

* Revert "removed if self.buffer.is_allocated() in realized (#14836)"

This reverts commit 72cf603805.
 2026-02-21 16:33:55 +09:00
qazal
5b6fcd1cda
 		gemm/asm: smallest cdna4 asm gemm test (#14925) 2026-02-21 11:56:05 +09:00
George Hotz
df7774661a
 		remove late numbering of UOps (#14923) 
* remove late numbering of UOps

* stupid fix

* dead code
 2026-02-21 09:18:48 +08:00
chenyu
24286c5593
 		fix clone for multi (#14919) 
also update empty_like to make sure it's backed by buffers
 2026-02-20 17:21:09 -05:00
chenyu
1fc1508f67
 		add assign to test_realize_is_realize.py (#14918) 2026-02-20 16:48:01 -05:00
chenyu
a4634b253a
 		fix empty_like for sharded tensor (#14915) 2026-02-20 16:30:04 -05:00
Nicolas Pinto
aa905db7f7
 		ptx: use setp.neu for float CMPNE (#14805) 
* ptx: use setp.neu for float CMPNE

* test ptx float CMPNE renders setp.neu

* check NaN behavior, not grep ptx strings...

* skip WEBGPU for test_cmpne_nan (Vulkan NaN behavior)

---------

Co-authored-by: Nicolas Pinto <41171+npinto@users.noreply.github.com>
Co-authored-by: chenyu <chenyu@fastmail.com>
 2026-02-20 16:11:04 -05:00
George Hotz
2611907afb
 		start ripping out old scheduler -- no maps (#14909) 
* start ripping out old scheduler -- no maps

* no more metadata
 2026-02-20 21:05:04 +08:00
nimlgen
1b3b94a72a
 		fix mockam mypy (#14908) 2026-02-20 15:15:05 +03:00
George Hotz
55d3a5def9
 		preallocate all realized buffers (#14823) 
* preallocate all realized buffers

* contiguous

* work

* comment that out

* move to schedule

* better

* correct fix

* just buffer

* disk bufs

* fixes disk tensor stuff

* fix symbolic stuff

* fix multi

* 162 failures

* bugfixes

* don't check that anymore

* fix schedule tests

* mnist should be contiguious

* type and buffer

* fix tests

* shrink axis correction

* mypy fixes

* tests skips

* same 37 failures

* dedup

* no shrink in the graph

* 29 failures

* skips

* fix custom kernel

* fix training

* those optimizations aren't supported currently

* simpler

* more correct

* tests

* 14 failures

* works

* fix that test

* broken

* 11 failures

* only kernel counts left

* fixes

* all tests pass

* remove tensor_map

* op test

* 200 -> 230

* test fixes

* fixes

* revert test_tiny thing

* guard

* revert that

* test tiny passes

* no contigs there

* base realize back

* Revert "no contigs there"

This reverts commit c45bb9fcfd.

* revert that

* chop many assigns

* 12 failures

* fix tests

* tests

* apply after

* pre-commit

* remove old code

* delete that

* fix types

* remove extra contig

* fix dataloader

* torch fix

* disk fix

* update kernel fusion numbres

* runs on amd

* restore kernel count

* add that rule back

* that

* disable that

* wrong

* add the correct rule for that folding

* more tests

* guard c1.arg

* no newlines

* realize those

* split into a different file

* remove detach/contig back

* skip 2

* update that
 2026-02-20 20:05:54 +08:00
nimlgen
dbf894215a
 		init mockam (#14889) 
* mockam

* more tests

* linter

* x
 2026-02-20 14:09:11 +03:00
George Hotz
6610255654
 		add the correct rule for gcd div/mod folding (#14905) 
* add the correct rule for that folding

* more tests

* guard c1.arg
 2026-02-20 18:11:54 +08:00
George Hotz
a28fc2fba7 hotfix: remove wrong symbolic rule 2026-02-20 17:09:18 +08:00
qazal
16ae96fa58
 		finish rdna4 sqtt (#14903) 
* unskip

* it's a wave pair in rdna4

* work

* that

* hidden archive

* generic s_delay, mystery InstOpRDNA4.UNK_60

* branch failing test

* UNK_60 is OTHER_VMEM_STORE

* rdna4 has both s_delay_alu and s_wait_alu

* real branch failing test

* rdna4 doesn't have JUMP_NO, it's NEXT with a flag for no jump

* make inst_delay skips recursive

* all rdna4 tests pass

* simm16 unwraps

* that has a name
 2026-02-20 16:06:13 +09:00
qazal
52b51a0324
 		test fixes from rdna4 sqtt (#14902) 2026-02-20 14:42:33 +09:00
qazal
32f569b573
 		viz/sqtt: decoder fixes pre rdna4/cdna4 work (#14900) 
* viz/sqtt: decoder fixes pre rdna4/cdna4 work

* fix

* branch_inst + more tests

* smaller
 2026-02-20 12:10:15 +09:00
qazal
e9ae3da711
 		viz: click on CALL node goes to codegen (#14609) 
* viz: click on CALL node goes to codegen

* colored name
 2026-02-20 11:13:11 +09:00
George Hotz
fc5677c28b
 		resnet dataloader + more test cleanups (#14899) 
* resnet dataloader

* tests
 2026-02-20 10:05:47 +08:00
chenyu
b9744ab62b
 		one more test_gpudims test (#14898) 
failure from the bad simplification attempt
 2026-02-19 18:18:44 -05:00
chenyu
9d6cf00be2
 		fix gpudim bug and test_split_2d_to_3d (#14896) 2026-02-19 16:46:24 -05:00
chenyu
2b31823ef9
 		update test_gpudims to prove bijectivity (#14895) 
* update test_gpudims to prove bijectivity

* one more
 2026-02-19 16:18:59 -05:00
chenyu
19ce7a3f7f
 		use z3 to verify gpudims output index (#14894) 
found a bug with z3
 2026-02-19 15:24:38 -05:00
chenyu
52f727738b
 		move test_grouped_dims to test/null (#14893) 
it's a pure helper
 2026-02-19 14:50:53 -05:00
chenyu
7400362a86
 		remove UOp.vars [pr] (#14891) 2026-02-19 12:09:39 -05:00
chenyu
06ef8a26b7
 		add a test case that triggers CALL passthrough_multi (#14887) 2026-02-19 10:45:40 -05:00
Kartik Vashishta
9a9c7648e9
 		system: fix pci_scan_bus vendor filter (#14885) 
* system: fix pci_scan_bus vendor filter

* fix: formatting
 2026-02-19 17:23:32 +03:00
George Hotz
f6c1cf343c
 		new symbolic rule from prealloc_bufs (#14883) 
* new symbolic rule from prealloc_bufs

* optim
 2026-02-19 20:57:30 +08:00
qazal
911399bee5
 		assembly/amd: move the kernel capture stuff out of helpers (#14881) 2026-02-19 16:28:48 +09:00
George Hotz
2f0f8b5776
 		more test relaxations from prealloc_bufs (#14880) 2026-02-19 14:23:28 +08:00
George Hotz
ab61c16730
 		fixes and test relaxations from prealloc_bufs (#14875) 
* fixes and test relaxations from prealloc_bufs

* fix error type and guard _mop

* revert that

* contiguous makes extra/torch_backend/test_kernel_fusion.py fail
 2026-02-19 11:37:25 +08:00