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
qazalandGitHub 7733c217c5 remove spam comments in test_schedule (#12224) 2025-09-17 18:24:55 +03:00
qazalandGitHub d917895569 map out rangeify errors in test_schedule (#12211) 
* map out rangeify errors in test_schedule

* skip that

* add to ci
 2025-09-17 09:10:28 +03:00
Sieds LyklesandGitHub 158506b91e Upgrade some divmod folding for symbolic divs (#12216) 
* use const_factor() instead of arg

* add test

* change div min_max

* add tests

* add divide_by_symbolic_gcd

* add tests

* one more test

* Slice to unbind symbolic

* deal with const factor properly

* minor cleanup

* divide_by_symbolic_gcd becomes UOp.gcd and UOp.divide_exact

* add tests

* add gcd_without_const

* fix divide_exact bug

* add factor_remainder

* add tests

* fix imports

* elif -> if

* remove expectedFailure

* add more tests

* add more unwrap

* fix signature of pop_const

* remove that

* remove that
 2025-09-17 03:00:50 +02:00
Sieds LyklesandGitHub 328bfe6b9b fix map_expand for symbolic shapes (#12218) 
fix incorrect default argument in resolve
 2025-09-17 01:20:18 +02:00
chenyuandGitHub 5b12764b83 add arange cat arange test (#12217) 
simple test case to catch wrong reduce const folding. also clean up the old arange complexity test
 2025-09-16 17:12:32 -04:00
nimlgenandGitHub 53655a4ee5 cuda: cleanup old comment (#12215) 2025-09-16 23:11:32 +03:00
chenyuandGitHub 6b808c5fe6 update TestSymbolicJit.test_plus1_pad (#12214) 
was failing because movement was not captured
 2025-09-16 15:57:50 -04:00
Shun UsamiandGitHub 2a72b00679 Add test for 2D tensor indexing in setitem (#12193) 
* Add test for 2D tensor indexing in setitem

* Fix _masked_setitem to handle multi dim indexing correctly

* Fix indent

* Add fuzz test for 3D tensor indexing in setitem

* Skip indexing fuzz test (slow)
 2025-09-16 14:57:25 -04:00
chenyuandGitHub c7b03457d7 Revert "Revert "more llvm intrinsics (#11961)" (#12194)" (#12195) 
This reverts commit df1c183e46.
 2025-09-16 14:55:31 -04:00
chenyuandGitHub 494bb12500 skip slow cifar bf16 on red benchmark (#12213) 
very slow to compile the fake bf16
 2025-09-16 14:55:01 -04:00
chenyuandGitHub 419e997187 increase benchmark timeout (#12212) 
account for compile cache, and it's annoying that job died due to timeout also messes the machine
 2025-09-16 14:09:02 -04:00
chenyuandGitHub 84d2d047ea Tensor.pad_to and Tensor.shrink_to (#12210) 
most of the time i want this instead of spelling out the args

also add more input validation to shrink
 2025-09-16 12:24:55 -04:00
qazalandGitHub 122a50fe8c assert kernel count (#12205) 2025-09-16 14:24:39 +03:00
chenyuandGitHub e555748807 test rangeify const folding (#12200) 
* test rangeify const folding

reduce i know how to fix, multi and test_cast_padded tbd

* test_instancenorm_3d is very slow
 2025-09-15 20:03:48 -04:00
chenyuandGitHub f732f66709 rangeify test_nn almost pass (#12198) 
* rangeify test_nn almost pass

* issue with jit

* flaky
 2025-09-15 17:49:20 -04:00
chenyuandGitHub 82e037aad5 ci test.yml updates (#12197) 
* ci test.yml updates

move docs together and external_benchmark_schedule to unit

* torch
 2025-09-15 17:09:02 -04:00
chenyuandGitHub 146c31586d split RANGEIFY ci (#12196) 
one CPU and one CL for speed
 2025-09-15 15:41:10 -04:00
chenyuandGitHub df1c183e46 Revert "more llvm intrinsics (#11961)" (#12194) 
This reverts commit d01e3d7719.
 2025-09-15 13:56:43 -04:00
b1tgGitHubb1tg
 		d01e3d7719 more llvm intrinsics (#11961) 
* more llvm intrinsics

* assert nan

* skip test_log_nan on metal

---------

Co-authored-by: b1tg <[email protected]>
 2025-09-15 13:05:23 -04:00
nimlgenandGitHub b63bd02969 update runtime docs (#12191) 2025-09-15 17:46:20 +03:00
qazalandGitHub 57e8bf61e8 viz: fix Specificity for rect styling (#12190) 2025-09-15 17:33:37 +03:00
chenyuandGitHub 72e010d816 fix rangeify ci (#12189) 
CL=1, and multitensor needs to test with CPU since CL does not support multi in CI
 2025-09-15 10:24:57 -04:00
qazalandGitHub f1bd06134d test fuse with RANGEIFY=2 (#12187) 2025-09-15 15:51:23 +03:00
qazalandGitHub ef0ef705fe viz: remove async from event listener (#12186) 2025-09-15 15:08:28 +03:00
qazalandGitHub d8855ec266 viz/serve.py cleanups (#12185) 
* don't assign unused variable

* *path to
 2025-09-15 13:43:26 +03:00
qazalandGitHub b8a74c1569 cpu: add disassembler err message (#12184) 
* cpu: add disassembler err message

* print msg
 2025-09-15 13:29:44 +03:00
qazalandGitHub a388d2cb1a remove PROFILE=1 option, it's just VIZ=1 [pr] (#12176) 
* remove PROFILE=1 option, it's just VIZ=1 [pr]

* sqtt

* sqtt 2

* return last

* rename
 2025-09-15 12:51:50 +03:00
George HotzandGitHub 65397bfdeb set testpath on pytest (#12183) 2025-09-15 16:13:05 +08:00
George HotzandGitHub ae0edc8a67 renumber ranges (#12182) 
* enable rangeify const folding

* renumber ranges for kernel deduping
 2025-09-15 13:03:39 +08:00
hoovedandGitHub e1fef895b1 don't hardcode weights path (#12171) 2025-09-15 00:33:47 -04:00
hoovedandGitHub 3a9db08b49 download data and ckpts for sd train/eval (#12170) 2025-09-15 00:31:45 -04:00
chenyuandGitHub bdb3afd566 failed test case for symbolic pad (#12179) 2025-09-15 00:25:21 -04:00
George HotzandGitHub 9fcc87761e enable rangeify const folding (#12181) 2025-09-15 12:02:19 +08:00
George HotzandGitHub 1353250b6c tags on bufferize are the tensor tags (#12180) 2025-09-15 11:46:03 +08:00
George HotzandGitHub 60d7db093e delete bufferized consts + output noops (#12163) 
* bring const folding to rangeify

* comment that
 2025-09-15 11:07:44 +08:00
qazalandGitHub 525c20dc7e viz: remove unused runtime_stats feature (#12177) 2025-09-15 02:53:05 +03:00
qazalandGitHub 75ff9b7a9a viz: add buffer lifetime to tooltip (#12175) 2025-09-15 02:33:50 +03:00
chenyuandGitHub 15b166ce6d bump test_module_runs to 30 seconds (#12174) 
25 seconds sometimes
 2025-09-14 16:48:40 -04:00
ttomsaGitHubSieds Lykles
 		943236ef74 move cast pat out of symbolic_simple (#11945) 
* move pat

* move it here

* rm extra check

---------

Co-authored-by: Sieds Lykles <[email protected]>
 2025-09-14 21:39:48 +02:00
Steven ShiandGitHub 25b1bc8eff added top k sampling to examples/mamba (#12061) 2025-09-14 15:27:34 -04:00
Shun UsamiandGitHub 34a05b31fe Fix advanced tensor indexing setitem (#12128) 
* Add failure test case for advanced tensor indexing setitem

* Fix advanced tensor indexing setitem when permuted

* Reduce line count

* Revert unnecessary change

* Combine two lines into one
 2025-09-14 15:22:40 -04:00
chenyuandGitHub d09c0f28c5 increase test_module_runs (#12173) 
timed out on ci windows llvm
 2025-09-14 15:19:21 -04:00
chenyuandGitHub 12a910f1d2 update torch 2.8 (#12172) 
support _reshape_alias. something is wrong with one case of unfold
 2025-09-14 15:19:03 -04:00
chenyuandGitHub 98ecab7563 remove ml_dtypes (#12169) 2025-09-14 14:20:05 -04:00
qazalandGitHub 02054b53fe remove tests that pre date the uop spec (#12168) 
* remove tests that pre date the uop spec

* const src

* for RANGEIFY=1

* update with bind

* remove import
 2025-09-14 18:47:42 +03:00
qazalandGitHub 1591e4f66b update outbufs selection in test_linearizer [pr] (#12166) 2025-09-14 13:46:49 +03:00
nimlgenandGitHub d1ae30f7ef hcq: do not spam with errors in -m device (#12150) 
* hcq: do not spam with errors in -m device

* um?

* um?

* nn

* helps?

* um?

* no gc?

* fix
 2025-09-14 10:56:59 +03:00
George HotzandGitHub d5bc27797b fix some multitensor on rangeify (#12162) 
* fix some multitensor on rangeify

* rangeify multi hacks

* copy on const
 2025-09-14 14:31:57 +08:00
Meng ZhuoandGitHub 4b7904eca9 add cpu support for riscv64 (#12136) 2025-09-14 11:40:58 +08:00
George HotzandGitHub bcafa72b7f use tags instead of graph_rewrite_map in rangeify (#12110) 
* use tags instead of graph_rewrite_map in rangeify

* new style, add realize

* metadata works

* simple failure

* fix

* loops

* stuff becomes a NOOP when you remove it

* stuff becomes a NOOP when you remove it

* tags on bufferize

* bmnist works

* locals don't work

* shippable

* fix some tests

* simpler map_realize

* remove const hack

* debuggable test

* broke

* assign test

* straight up bug

* wooo it passes

* sink shouldn't be there

* fix ops

* bmnist

* kv cache ish

* Set RANGEIFY context variable to 0

* should work normal

* better

* types

* hacks to fix test_symbolic

* pm_add_buffers

* tests should pass
 2025-09-14 11:39:01 +08:00