geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity
12,474 Commits 4 Branches 15 Tags
02da66053ee6a5e9739c22f6a5ae87ec6037bdc1
Commit Graph
5215 Commits
Author SHA1 Message Date
b1tgandGitHub a9ea36de79 assembly/amd: v_cmp_lg_f32 is ordered not-equal (#14982) 2026-03-03 15:37:48 +08:00
wozeparrotandGitHub c35de9bd68 asm_gemm: support more sharding (#15002) 2026-03-02 23:16:37 -08:00
chenyuandGitHub 5dcf29b1a0 use clone in test_swap_slices (#15096) 2026-03-02 22:05:12 -05:00
sirhcmandGitHub c70e8af068 move IMAGE FLOAT16 logic to allocations (#15095) 
* FLOAT16 logic in allocations

* cleanup

* separate that

* only apply when IMAGE == 1

* test passing now

* create image buffers earlier
 2026-03-02 22:00:05 -05:00
George HotzandGitHub d483e4153a buffer view is like buffer (#15082) 
* buffer view is like buffer

* fix

* swap_reshape_shrink

* contiguous on gguf, fix overlap

* revert that

* _device_supports_view

* this

* fix that test

* 0 buffers

* that test was wrong

* this

* check correct size

* contig BUFFER_VIEW

* this

* fix tests

* buffer view tests

* om

* fix torch

* no MOCKGPU

* skip
 2026-03-03 09:52:33 +08:00
chenyuandGitHub 14d1c5fdfd assign fusion tests on detach and contiguous_backward (#15092) 2026-03-02 15:21:51 -05:00
qazalandGitHub f7aeff6061 viz: cli.py cleanups, do not require PYTHONPATH (#15085) 
* cleanup the print

* sys.exit

* equal check

* cleanup unpacker

* cli doesn't need PYTHONPATH

* no semicolons

* %s/PYTHONPATH=. //g
 2026-03-02 19:24:38 +09:00
sirhcmandGitHub 977c270774 IMAGE=1 kernel count failing tests (#15083) 2026-03-02 04:35:26 -05:00
George HotzandGitHub 3539693555 Support triu variable on diagonal + SDPA symbolic (#15081) 
* triu variable

* fails

* dumbbb

* no commutative in reshape

* real fix

* revert that

* sdpa symbolic tests
 2026-03-02 12:19:48 +08:00
NickGitHubGeorge Hotz
 		8e8e9f6ff6 assert removal for _tri() + tests (#15073) 
* assert removal for _tri() and tests

* removed import

* tests triu/tril like in prefill

---------

Co-authored-by: George Hotz <[email protected]>
 2026-03-02 10:34:28 +08:00
nimlgenandGitHub ccbbca05ef beam: add dev_timeout for am (#15063) 
* beam: add dev_timeout for am

* all covered

* fk

* x

* fuzz

* reset

* f
 2026-03-01 16:57:29 +03:00
chenyuandGitHub 103ea16ec0 add contiguous back to svd (#15074) 
can cause infinite loop
 2026-02-28 16:49:26 -05:00
chenyuandGitHub fe0fa8333b Revert "improve Tensor.sort indices (#15070)" (#15072) 
This reverts commit e3003631f2.
 2026-02-28 14:40:30 -05:00
chenyuandGitHub e3003631f2 improve Tensor.sort indices (#15070) 
* improve Tensor.sort indices

instead of N^2 match at the end, have an arange to start and go through the same N(logN)^2 path

* contiguous
 2026-02-28 14:16:16 -05:00
chenyuandGitHub 76170d035a relax atol for test_xlm_roberta_large (#15066) 2026-02-28 11:22:35 -05:00
nimlgenandGitHub 9b3450c9da test gpu crash on cdna (#15062) 2026-02-28 13:17:59 +03:00
George HotzandGitHub bb84e389cf functions for llama trainer (#15045) 
* functions for llama trainer

* function there

* axis match

* fix multi

* lil cleaner

* there's a bug with HK_FLASH_ATTENTION

* training functions

* for commit
 2026-02-28 12:15:18 +08:00
chenyuandGitHub 151608aa90 update test_multiple_to_single_device (#15056) 
follow up to #14482, add SCACHE=0 to the test
 2026-02-27 21:44:33 -05:00
chenyuandGitHub 5fd06f4f02 differentiable setitem (#15054) 
* differentiable setitem

go through the where path for bw

* no return
 2026-02-27 17:25:15 -05:00
chenyuandGitHub db6b3e1edc fix mixed setitem with both basic and tensor indexing (#15050) 2026-02-27 15:35:48 -05:00
chenyuandGitHub c9f6d8751b don't remove_bufferize for Invalid (#15053) 
* don't remove_bufferize for Invalid

* replaced
 2026-02-27 15:16:09 -05:00
qazalandGitHub b8a55d5f68 sqtt: new packet types, add discovery script (#14960) 2026-02-28 04:27:27 +09:00
chenyuandGitHub 1406d49eef failed test cases for advanced setitem (#15048) 2026-02-27 10:50:18 -05:00
qazalandGitHub ad99b77f6d assembly/amd: add gfx12_asm_vflat llvm tests, disasm fixes (#15046) 
* add gfx12_asm_vflat.s

* work
 2026-02-27 20:20:31 +09:00
George HotzandGitHub 010d2790ce fix multi minimal (#15044) 2026-02-27 14:31:58 +08:00
George HotzandGitHub d23b79530e remove disk from GGUF GEMV test (#15041) 
* remove disk from GGUF GEMV test

* keep copy
 2026-02-27 12:03:00 +08:00
chenyuandGitHub d345f7f5dc remove _pending_assigns (#15040) 2026-02-26 22:38:10 -05:00
George HotzandGitHub 37e31e7da4 gguf gemv test (#15039) 
* add gemv tests

* gguf big

* skip

* make realize optional
 2026-02-27 10:54:43 +08:00
chenyuandGitHub 0f94a4bb73 failed test case for early fixup const copy (#15038) 
* failed test case for early fixup const copy

wrong with PAD

* test no copy
 2026-02-26 19:09:33 -05:00
chenyuandGitHub 3a4db53b43 raise RuntimeError in schedule for conflicted var_val [pr] (#15031) 2026-02-26 15:16:01 -05:00
George HotzandGitHub fe3ee8c27e fix symbolic shapes in calls (#15021) 
* fix symbolic shapes in calls

* fix after in the big graph

* real tests
 2026-02-26 17:17:18 +08:00
George HotzandGitHub 2655655a0c call gradient creates a call (#15020) 
* function creates a full subgraph

* tests

* fix var

* fix tests

* implict assign/contig

* move kv init
 2026-02-26 14:15:29 +08:00
chenyuandGitHub ed9d475a12 assign tests with test_function (#15015) 2026-02-25 16:15:59 -05:00
nimlgenandGitHub faa66e0a61 mi350 hive_reset am repro (#15014) 2026-02-25 21:30:18 +03:00
George HotzandGitHub 0d35b67f2c revert realize to only be buffers (#15008) 
* revert realize to only be buffers

* fix that

* broken attention

* Revert "broken attention"

This reverts commit a23c3cd96c.

* and that
 2026-02-25 22:43:06 +08:00
George HotzandGitHub 68831cd852 add more tests to test_function (#15003) 
* add more tests to test_function

* add function to llm

* function decorator on llm

* works

* symbolic fixups

* minimum change

* implicit inputs

* don't actually update llama yet
 2026-02-25 18:42:06 +08:00
George HotzandGitHub e3fa9896b7 start function and add walk rewrite (#14992) 
* start function and add walk rewrite

* work

* add function on feed_forward

* llm progress

* stuff

* none of that
 2026-02-25 13:56:27 +08:00
chenyuandGitHub fde7a40bb0 allow dtype mismatched assign on disk (#14993) 
reverted #14473, that was a bad idea. also added a test that safe_save only has copy
 2026-02-24 20:49:55 -05:00
chenyuandGitHub 5fd4fc0c6d fix tinyfs (#14974) 
* fix tinyfs

* fix that
 2026-02-24 08:50:53 -05:00
George HotzandGitHub 8a6dffc87e Tensor.callify will be the JIT (#14983) 
* close

* simple callify, support linear in the scheduler

* all tests pass

* everyone is happy

* dumb test

* Remove unnecessary blank line in rangeify.py
 2026-02-24 18:42:24 +08:00
George HotzandGitHub b643fca51e clean up complete_create_schedule_with_vars (#14980) 
* clean up complete_create_schedule_with_vars

* transform_to_call

* update viz tests
 2026-02-24 16:12:36 +08:00
chenyuandGitHub 0bda5585c7 unit test TestTinyFS (#14972) 
these passed before the allocation change
 2026-02-23 16:59:39 -05:00
imaoloGitHubnimlgen
 		405d37423e call release() in MetalAllocator._free (#14970) 
* add failing test

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

* Update test_metal.py

---------

Co-authored-by: nimlgen <[email protected]>
 2026-02-23 23:33:31 +03:00
chenyuandGitHub 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
ttomsaGitHubchenyu
 		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 <[email protected]>
 2026-02-23 10:31:36 -05:00
George HotzandGitHub 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
wozeparrotandGitHub 25565b2410 fa: test for mp (#14907) 2026-02-22 21:47:36 -08:00
qazalandGitHub 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 HotzandGitHub 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
qazalandGitHub 1538960002 viz: smaller view for repeated asm instructions in cfg (#14954) 
* simple test

* todo

* feature
 2026-02-23 10:41:43 +09:00