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

51 commits

Author SHA1 Message Date
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
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
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
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
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
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
George Hotz
fc5677c28b
 		resnet dataloader + more test cleanups (#14899) 
* resnet dataloader

* tests
 2026-02-20 10:05:47 +08:00
chenyu
52f727738b
 		move test_grouped_dims to test/null (#14893) 
it's a pure helper
 2026-02-19 14:50:53 -05: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
chenyu
0c85b93938
 		support shink sharded and non-sharded axes (#14874) 
simpler to just support it
 2026-02-18 20:54:10 -05:00
chenyu
8c830c5b44
 		test_full_like_shrink_on_shard_axis (#14870) 
* test_full_like_shrink_on_shard_axis

add a test case that triggers non-copy branch in mstack_early_shrink

* 0
 2026-02-18 19:23:44 -05:00
chenyu
f771de6738
 		gc.collect() to get the correct GlobalCounters.mem_used in tests (#14868) 
test can be flaky if gc happens in between
 2026-02-18 15:01:23 -05:00
chenyu
f84a11bb9f
 		delete uneven shard tests and mentions (#14867) 2026-02-18 14:10:33 -05:00
George Hotz
af839b2bd1
 		remove all the outerworld stuff, it was too complex (#14852) 2026-02-18 17:44:11 +08:00
chenyu
f147791105
 		update test to reset and test kernel_count directly (#14832) 2026-02-17 11:48:46 -05:00
chenyu
f2f039cc0f
 		fix chained full-buffer assign (#14828) 
this shows issue that pm_remove_bufferize drops tags, will fix in bufferize next. this also fixed rand being different in jit vs no-jit
 2026-02-17 09:11:04 -05:00
George Hotz
ff60dab622
 		Revert "big sink is on base (#14819)" (#14825) 
This reverts commit 5fc3d8109f.
 2026-02-17 19:18:06 +08:00
George Hotz
5fc3d8109f
 		big sink is on base (#14819) 
* big sink is on base

* contiguous fixes tests
 2026-02-17 18:32:56 +08:00
qazal
f590564bf7
 		gemm multiple is only for cdna4 asm (#14814) 
* gemm multiple is only for cdna4 asm

* move to backend

* and arch

* path
 2026-02-17 14:00:02 +09:00
chenyu
f290af6c7d
 		test_schedule always test with SPLIT_REDUCEOP=0 (#14802) 
* test_schedule always test with SPLIT_REDUCEOP=0

except tests that tests SPLIT_REDUCEOP=1

* like that
 2026-02-16 15:30:26 -05:00
Nicolas Pinto
20b658b786
 		fuse MULACC after MUL->SHL (#14788) 
* decompositions: fuse (x << n) + c to MULACC

MUL→SHL converts x*(2^n) to x<<n before MULACC can fuse (x*c)+y.
Add pattern to also fuse (x<<n)+c → MULACC(x, 2^n, c) for backends
that support both MULACC and SHL.

* test: add test_mulacc_shl for SHL->MULACC fusion

* test: relax test_mulacc_unrolled to >= 4

SHL->MULACC fusion now also catches power-of-2 address calculations,
increasing MULACC count from 4 to 6 on PTX. the test's intent is that
each unrolled multiply is individually fused (not grouped), so >= 4
is the correct assertion.

---------

Co-authored-by: Prithvish <deformercoding@gmail.com>
Co-authored-by: Nicolas Pinto <41171+npinto@users.noreply.github.com>
Co-authored-by: Nicolas Pinto <npinto@mbp23.local>
 2026-02-16 16:26:44 +08:00
qazal
8e7c5f5b09
 		remove Tensor.training = True in test_arange (#14781) 2026-02-16 11:19:42 +09:00
qazal
156b6cb7e4
 		native bf16 cast in cdna4 (#14574) 
* native bf16 cast in cdna4

* don't need contig backward

* simpler

* contig bw still wins in those cases
 2026-02-16 10:51:32 +09:00
chenyu
352845d8cc
 		update cast to uint tests (#14768) 
result in valid range should work, add intermediate cast to NIRRenderer since it's UB for [128, 256)
 2026-02-15 10:55:13 -05:00
qazal
ceccc8eb86
 		unskip now passing multi tests [pr] (#14759) 2026-02-15 20:30:00 +09:00
qazal
42b6bf0b7a
 		fix sdpa causal failing test on multi (#14762) 
* simple failing test

* device is from xq
 2026-02-15 16:54:33 +09:00
George Hotz
0e215c433d
 		remove hack from cast (#14760) 
* remove hack from cast

* skip tests

* linters to 3.12, another skip

* fix rand

* m_
 2026-02-15 13:56:38 +08:00
George Hotz
d176af6269
 		start outerworld call test, fix gate (#14758) 2026-02-15 12:35:01 +08:00
chenyu
ca68037f26
 		lazy basic setitem to unrealized Tensor (#14756) 
undo the view and make it a mask, this fuses the setitem with any pending compute too.

one behavior change is that for target not backed by a buffer (const and arange), rangeify makes output contiguous under the hood.
this is stricter better than raise and ask user to call contiguous, as that would no longer be fuse-able.
 2026-02-14 20:27:03 -05:00
chenyu
95f4c7e90a
 		fix limit_bufs to not limit index (#14751) 
index is not real buffer. also made MAX_KERNEL_BUFFERS a ContextVar
 2026-02-14 16:00:03 -05:00
chenyu
8f6772fd8c
 		more setitem kernel mem tests (#14749) 
* more setitem kernel mem tests

test only the slice is accessed

* update
 2026-02-14 11:01:03 -05:00
chenyu
446909fb7a
 		more setitem kernel tests (#14748) 
check where realize happened
 2026-02-14 09:57:46 -05:00
Christopher Milan
eaa9506a00
 		disallow subnormals in emulated test_dtype (#14744) 2026-02-14 00:11:57 -05:00
chenyu
dca7819f76
 		more setitem into unrealized tests (#14737) 
* more setitem into unrealized tests

into empty, const with alu, and arange

* typo
 2026-02-13 20:28:51 -05:00
chenyu
8b205a007e
 		lazy setitem for realized target (#14735) 2026-02-13 12:20:14 -05:00
Christopher Milan
08a555c875
 		skip test_expand_buffer_before_cast on WEBGPU metal (#14724) 2026-02-13 00:01:05 -05:00
Christopher Milan
c30bb0f006
 		fix WEBGPU isnan check (#14711) 2026-02-12 17:01:18 -05:00
nimlgen
b376bd7a21
 		jit: fix raw in same kernel (#14699) 
* jit: fix raw in same kernel

* fix

* ugh

* x

* simpler
 2026-02-12 15:33:32 +03:00
George Hotz
095a064ba8
 		test.yml explicitly says backend (#14700) 
* test.yml explicitly says backend

* 1e-5
 2026-02-12 16:03:44 +08:00