nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
11,429 Commits 334 Branches 15 Tags
execution_unit
Commit Graph
11429 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
b1tgGitHubb1tg
 		6d53cac457 dtype fuzz: log need input > 0 (#11979) 
Co-authored-by: b1tg <[email protected]>
 2025-09-03 12:10:42 -04:00
Jordan ChalupkaandGitHub 68e83b850f nbytes should raise an exception when size is unlimited (#11928) 
* nbytes should raise an exception when size is unlimited

* adding a test
 2025-09-03 07:06:20 -07:00
Sieds LyklesandGitHub 86e908db57 cast parents of int64 alu to int32 if possible (#11977) 
* add overflows helper

* add rules

* x -> y

* check overflow of u too

* cleaner

* use alu instead of replace to preserve vectorization

* just one rule

* add test
 2025-09-03 11:05:04 +02:00
Sieds LyklesandGitHub 033184b3cb parse_valid with non const rhs (#11957) 
* const to using vmin/vmax

* add test

* convert to int

* remove left over part of and
 2025-09-03 08:08:46 +02:00
Sieds LyklesandGitHub 53eff8970a add Ops.GEP to _min_max (#11976) 2025-09-03 07:07:54 +02:00
Sieds LyklesandGitHub d1d0960e6e remove intermediate cast using bounds - weaker pattern (#11974) 2025-09-03 06:24:40 +02:00
Sieds LyklesandGitHub 8a2846b31a assert embedding input is integer dtype (#11963) 
* cast embedding input

* raise error if not using int for index embedding
 2025-09-03 01:44:26 +02:00
wozeparrotandGitHub d16cc6c012 feat: resume ckpt (#11970) 2025-09-02 15:47:48 -07:00
George HotzandGitHub 1b73993521 pyrender to render uops (#11968) 
* pyrender to render uops

* new pyrender style

* pyrender works

* list str

* store render
 2025-09-02 15:44:01 -07:00
chenyuandGitHub e921fb44ee clean up testnvidia env (#11969) 2025-09-02 18:29:00 -04:00
chenyuandGitHub 69dd1817d0 raise RuntimeError in merge_dicts instead of assert [pr] (#11965) 2025-09-02 17:18:44 -04:00
qazalandGitHub f750c15965 viz: add python marker (#11952) 
* viz: add python marker

* remove duplicate
 2025-09-02 23:44:00 +03:00
George HotzandGitHub 550cf2ca7f tests from postopt (#11964) 
* tests from postopt

* reraise is fine
 2025-09-02 13:34:17 -07:00
qazalandGitHub b977ec0813 viz: axes domains cleanup (#11962) 2025-09-02 19:30:45 +03:00
nimlgenandGitHub 897254ad6c ci: add dev<->cpu copy speeds (#11959) 2025-09-02 15:22:44 +03:00
George HotzandGitHub 74040663bf make ptrdtype a UOp property (#11955) 2025-09-01 16:35:43 -07:00
George HotzandGitHub 0dfca4e74b add failing test for rangeify setitem (#11954) 2025-09-01 16:24:35 -07:00
wozeparrotandGitHub 7c21271a5f feat: end_lr envvar (#11953) 2025-09-01 14:53:07 -07:00
chenyuandGitHub 6a40216724 correct bf16 fuzz input in test_dtype_alu (#11933) 
it was using float16 inputs, now it's uint16 then convert to bf16
 2025-09-01 10:52:26 -04:00
chenyuandGitHub 965ea59b16 test_dtype_alu use AMD_LLVM from helpers (#11950) 2025-09-01 10:03:17 -04:00
b1tgGitHubb1tgchenyu
 		a9f07c31bc fix amd llvm sqrt (#11936) 
* fix amd llvm sqrt

* lint

---------

Co-authored-by: b1tg <[email protected]>
Co-authored-by: chenyu <[email protected]>
 2025-09-01 09:31:14 -04:00
qazalandGitHub 0a53e72f70 viz: fix trace duration in python test decoder (#11949) 2025-09-01 14:32:25 +03:00
qazalandGitHub 27c9ed5a84 viz: more consistent naming of events (#11948) 
* s/shapes/events in test_viz

* s/bufs/events in the memory packer
 2025-09-01 14:16:47 +03:00
qazalandGitHub c7bb561ef9 remu: add v_rsq_f32_e32 instruction (#11947) 
https://github.com/tinygrad/tinygrad/pull/11936 introduces a change to
the AMD LLVM renderer that outputs this instruction. Adding both 32 and
64 bit variants.
 2025-09-01 11:29:31 +03:00
Sieds LyklesandGitHub d9560a631c remove cast between ints if safe (#11946) 2025-09-01 05:56:49 +02:00
Sieds LyklesandGitHub a19d689481 fix vec dtype _min_max (#11944) 2025-09-01 03:24:07 +02:00
Sieds LyklesandGitHub f32f3464d6 Can safe cast from certain ints to floats (#11941) 
* add rule

* add some tests

* prevent infinite loop with bfloat16

* add some ints to double and float can_safe_cast

* add tests
 2025-09-01 00:51:24 +02:00
Sieds LyklesandGitHub 1c6e43c203 Double cast is one cast if intermediate cast is safe (#11939) 
* add rule

* add some tests

* prevent infinite loop with bfloat16

* prevent more infinite rewrite
 2025-09-01 00:36:29 +02:00
wozeparrotandGitHub 7e68045fb2 feat: small llama3 training (#11829) 2025-08-31 13:41:47 -07:00
nimlgenandGitHub 020abe0556 hcq: finalize without synchronization when in error state (#11872) 
* hcq: finalize without synchronization when in error state

* ooops

* fix

* fix

* fix
 2025-08-31 18:39:13 +03:00
qazalandGitHub 2004c9757d tracing: add default clock (#11935) 2025-08-31 18:24:44 +03:00
b1tgGitHubb1tg
 		c1eeb3b99c only skip AMD_LLVM (#11934) 
Co-authored-by: b1tg <[email protected]>
 2025-08-31 18:15:47 +03:00
b1tgGitHubb1tg
 		75d380a77c fix transcendentals in python renderer (#11932) 
* fix transcendentals in python renderer

* add test

---------

Co-authored-by: b1tg <[email protected]>
 2025-08-31 09:37:17 -04:00
Sieds LyklesandGitHub 61e4dc6ad5 render special arg in cstyle if arg is UOp (#11931) 2025-08-31 07:01:29 +02:00
Sieds LyklesandGitHub d3252ccd85 fix special vmax when arg is UOp (#11930) 2025-08-31 06:54:39 +02:00
qazalandGitHub 0bacd9fc9b viz: give disassembly its own node (#11927) 2025-08-31 00:28:52 +03:00
chenyuandGitHub af89be317e relax rtol for bfloat16 test_dtype_alu (#11926) 2025-08-30 17:16:08 -04:00
George HotzandGitHub 632c2fb119 lowerer works on rangeifed + print exception (#11925) 2025-08-30 12:05:44 -07:00
qazalandGitHub c27b99d68f viz: refactor to indexed rewrite traces (#11923) 2025-08-30 20:01:47 +03:00
qazalandGitHub 9aff00a6ea switch viz command line args to pathlib (#11922) 2025-08-30 18:13:47 +03:00
qazalandGitHub c86ee5bfaf viz: canonicalize device name colors (#11921) 2025-08-30 18:12:30 +03:00
nimlgenandGitHub a4f05ebd1a ci: rebuild gpuocelot with boost libs (#11920) 2025-08-30 17:24:19 +03:00
qazalandGitHub bf0d055b39 viz: color by name (#11919) 2025-08-30 16:04:58 +03:00
Sieds LyklesandGitHub 0bc34c000f simplify range mod its own upper bound (#11917) 
* add rules

* add tests
 2025-08-30 08:37:35 +02:00
chenyuandGitHub 561318fea7 Tensor.cos in test_stype_alu (#11916) 
* Tensor.cos in test_stype_alu

* need this fix anyway
 2025-08-29 20:26:36 -04:00
NoahKusabaGitHubchenyu
 		0838021753 remove np from beautiful_cifar (#10988) 
* remove np from beautiful_cifar

* remove np from cifar

* rename variable and rename tensor.arrange to just tensor.randperm

---------

Co-authored-by: chenyu <[email protected]>
 2025-08-29 19:34:16 -04:00
nimlgenandGitHub cf9d8c8142 ci: pin boost for macos runners (#11910) 2025-08-30 01:38:06 +03:00
nimlgenandGitHub c6e342cdac mockgpu: no hang if gpuocelot failed (#11915) 2025-08-30 00:44:49 +03:00
chenyuandGitHub 26d03a86a1 test_symbolic_ops.py cleanup (#11895) 2025-08-29 17:11:59 -04:00
b1tgGitHubb1tg
 		b2cc06218a python bfloat16 (#11912) 
* python bf16

* _to_torch_storage_type

---------

Co-authored-by: b1tg <[email protected]>
 2025-08-29 15:18:02 -04:00