nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
14,107 Commits 334 Branches 15 Tags
dd16d5aead62e0207c0c3c50c19bc8b67e176c55
Commit Graph
1838 Commits
Author SHA1 Message Date
chenyuandGitHub 23e9e76e8c DEFAULT_FLOAT/DEFAULT_INT ContextVar [pr] (#17265) 2026-07-28 19:08:18 -04:00
George HotzandGitHub 7d48926293 switch _device_num to AxisType.DEVICE range (kimi) (#17252) 
* switch _device_num to AxisType.DEVICE range (kimi)

* improve

* don't exclude device for ranges

* current multi ends the range
 2026-07-28 13:52:47 -07:00
chenyuandGitHub 749e002286 remove some dtype= when construct UOp [pr] (#17257) 
prep for removing the dtype field
 2026-07-28 16:05:35 -04:00
chenyuandGitHub c9e11544df delete explicit casts [pr] (#17255) 
handled by broadcast
 2026-07-28 13:57:13 -04:00
chenyuandGitHub 37cf15942b UOp.const(dtype=None) infers from from_py [pr] (#17253) 
eventually we drop the dtype arg once everything const dtype is inferred
 2026-07-28 12:06:55 -04:00
chenyuandGitHub 1380d6cc5a pm_long_decomp doesn't depend on operand dtype [pr] (#17249) 
* pm_long_decomp doesn't depend on operand dtype [pr]

the const the rules returned won't have strong dtype

* fix
 2026-07-28 01:15:04 -04:00
chenyuandGitHub f837ca3587 don't match const dtype in UPat [pr] (#17247) 2026-07-27 22:50:08 -04:00
chenyuandGitHub 654d475d37 update pm_long_decomp [pr] (#17245) 
* update pm_long_decomp [pr]

instead of reading dtype from UOp (which might be const that's going to be weak), just pass output dtype in tag

* cleanup tag
 2026-07-27 22:28:50 -04:00
chenyuandGitHub 896afad9bf don't match strong typed const in UPat [pr] (#17240) 2026-07-27 16:19:01 -04:00
chenyuandGitHub 818a892ebc flip from_py to use weak dtypes [pr] (#17229) 2026-07-27 10:07:00 -04:00
chenyuandGitHub eb889053bf weak frompy prerequisite [PR] (#17213) 2026-07-25 22:35:45 -04:00
sirhcmandGitHub 9fdaa4bff2 standardize Program class (#17189) 2026-07-25 00:17:38 -04:00
chenyuandGitHub 0a6125e571 Invalid is bool, put weakint in promo lattice [pr] (#17188) 2026-07-25 00:06:53 -04:00
chenyuandGitHub 86b42d7cb8 clean up unbroadcast [PR] (#17182) 
also renamed
unbroadcast -> pm_expand_broadcast,
broadcast_binary -> expand_broadcast
 2026-07-24 11:48:07 -04:00
chenyuandGitHub 2dd2e3d786 don't support Tensor(list(np.array)) [pr] (#17175) 2026-07-24 03:13:28 -04:00
chenyuandGitHub a292dab8ab corner cases from weak const branch (#17173) 2026-07-24 00:09:56 -04:00
chenyuandGitHub 0f9edb02f6 don't use promo_lattice in fast_idiv [PR] (#17165) 
int64 will link to weakfloat
 2026-07-23 19:20:53 -04:00
chenyuandGitHub 723309b5c0 update hand_coded_optimizations [pr] (#17144) 
only upcast when they share valids
 2026-07-22 22:26:27 -04:00
chenyuandGitHub 2aafca411d Revert "reorder sym in full_rewrite_to_sink [pr] (#17141)" (#17142) 
This reverts commit 7755480f46.
 2026-07-22 19:47:47 -04:00
chenyuandGitHub 7755480f46 reorder sym in full_rewrite_to_sink [pr] (#17141) 
* reorder sym in full_rewrite_to_sink [pr]

* fix
 2026-07-22 19:41:00 -04:00
chenyuandGitHub a7c693d2fd rework pm_lower_weakint [pr] (#17136) 
deleted pm_no_index and many "remove hanging casts"
 2026-07-22 15:53:11 -04:00
George HotzandGitHub 39924387b1 LOOP is srcless RANGE (kimi) (#17129) 
* LOOP is srcless RANGE (kimi)

* upd

* cleanups

* cleanups

* we don't float anymore
 2026-07-21 18:41:31 -07:00
nimlgenGitHubgeohot
 		b764599d87 add Ops.LOOP + conditional Ops.END (kimi) (#17117) 
* add Ops.LOOP + conditional Ops.END (kimi)

* c

* x

---------

Co-authored-by: George Hotz <[email protected]>
 2026-07-21 22:54:14 +03:00
chenyuandGitHub 2864036e8e correct more spelling of coalesce (#17107) 2026-07-20 21:55:37 -04:00
sirhcmandGitHub f3a5337825 correct spelling of coalesce (#17103) 2026-07-20 21:44:04 -04:00
George HotzandGitHub 2a81616492 update rules for INDEX mops (#17105) 2026-07-20 18:43:00 -07:00
chenyuandGitHub 13ca9bd8a6 remove dtypes.index again (#17104) 
also reverted some dtype change, the split made things needlessly complicated
 2026-07-20 20:30:04 -04:00
chenyuandGitHub 4b8db13e01 rdna int8 wmma (#17098) 
nice to fix _wmma_name, also more generic tests
 2026-07-20 17:00:50 -04:00
chenyuandGitHub 89117d8b9e use real shift in l2i decomp [pr] (#17080) 
works for variable shift distace too, also fixed signed arithmetic fill
 2026-07-19 13:15:06 -04:00
George HotzandGitHub d8b83daac6 set tc_upcast_axes to None when done with it (#17053) 
* set tc_upcast_axes to None when done with it

* no tag needed
 2026-07-16 14:15:21 -07:00
George HotzandGitHub 2b1146b3f4 further clean up wmma (#17048) 
* further clean up wmma

* comment
 2026-07-16 11:43:23 -07:00
George HotzandGitHub 61e104bdfb use UOp.wmma everywhere (#17045) 
* use UOp.wmma everywhere

* fix
 2026-07-16 10:40:48 -07:00
chenyuandGitHub c23d8188e1 remove _ensure_float [pr] (#17030) 
do this cast late. allow `SQRT(int)`
 2026-07-15 11:19:20 -04:00
George HotzandGitHub 246eb51523 more dtype.count removal from x86 and others (#16989) 
* switch x86 to use numel instead of dtype.count

* works?

* remove dtype.count from x86 and others
 2026-07-12 09:10:02 -07:00
chenyuandGitHub 047a467bf9 delete UOp._stack and UOp.vectorize [PR] (#16988) 2026-07-11 15:26:55 -04:00
George HotzandGitHub 77823056d4 remove dead DSP code and pre_matcher (#16968) 2026-07-10 11:27:50 -07:00
George HotzandGitHub d80c971ea2 param + cast dtype in arg (#16955) 
* param + cast dtype in arg

* require dtype

* fix replace in dtype decomps

* fix x86

* dtype none is fine there

* Revert "dtype none is fine there"

This reverts commit 5315cc3e59.
 2026-07-09 16:32:01 -07:00
George HotzandGitHub 1df49a7bf5 dtype doesn't need to be explicit in most places now (glm) [PR] (#16951) 
* dtype doesn't need to be explicit in most places now (glm) [PR]

* fix amd pcode

* image idx long/int meh

* real image fix
 2026-07-09 14:41:34 -07:00
chenyuandGitHub ba2c68b1ed start dtypes.index [PR] (#16949) 2026-07-09 15:12:23 -04:00
chenyuandGitHub 8b96b95af5 tighter pm_no_weakints [PR] (#16946) 
towards spliting weakint and index
 2026-07-09 14:31:16 -04:00
nimlgenandGitHub c9f6b2e42b hcq2: prereq for linker (#16945) 2026-07-09 17:21:21 +03:00
George HotzandGitHub 00d01d978d add dtype production rule checked with SPEC=2 (#16939) 
* add dtype production rule checked with SPEC=2

* no None support yet

* detach passes through

* work

* fix

* line len
 2026-07-08 18:04:34 -07:00
chenyuandGitHub e2fc928d85 use get_idx and get_valid in coalese [PR] (#16937) 2026-07-08 17:20:54 -04:00
George HotzandGitHub fdffc6c0c8 remove dtypes base (#16931) 
* remove dtypes base

* find/replace bug

* vcount is junk too
 2026-07-08 11:48:11 -07:00
George HotzandGitHub 63cb1369cb remove ImageDType (glm) (#16918) 
* remove ImageDType (glm)

* gpt work

* better

* is_image_shape

* cleanups

* remove crap from index

* revert crap python change

* image index with float, this needs to be fixed

* simp
 2026-07-08 11:27:52 -07:00
George HotzandGitHub 6358f939e7 param arg is always the shape (#16914) 
* param arg is always the shape

* earlier

* pm_no_weakints

* real fix
 2026-07-08 10:51:56 -07:00
George HotzandGitHub 6e05cbdbcb remove the dtype.scalar calls everywhere [pr] (#16930) 
* remove the dtype.scalar calls everywhere

* remove base everywhere

* Revert "remove base everywhere"

This reverts commit 0dd0a5a243.
 2026-07-08 10:10:03 -07:00
George HotzandGitHub e4d0d634d4 remove a bunch of dtype.count [PR] (#16929) 
* remove a bunch of dtype.count

* remove nonsense rules

* rm dead code
 2026-07-08 09:34:46 -07:00
RaineGitHubGeorge Hotz
 		4f7d4e95d7 fix <= 2 case (#16923) 
Co-authored-by: George Hotz <[email protected]>
 2026-07-07 18:17:58 -07:00
George HotzandGitHub 4a51047146 remove unused pm_syntactic_sugar (#16920) 
* remove unused pm_syntactic_sugar

* merge in group for reduce

* merge in add loads

* that ctx is unused

* late loads

* Revert "late loads"

This reverts commit ee70cffcee.
 2026-07-07 15:40:52 -07:00