sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 1 Actions 23 Activity
14,176 Commits 334 Branches 15 Tags
http_threading
Commit Graph
1310 Commits
Author SHA1 Message Date
chenyuandGitHub 6c0ec39279 UOp.is_invalid [PR] (#17351) 
helper to prep ConstArg
 2026-08-01 02:36:32 -04:00
chenyuandGitHub b502fc1367 more const arg -> val (#17350) 2026-08-01 02:18:10 -04:00
George HotzandGitHub a88f832f0c remove UOp.val (#17345) 2026-07-31 18:58:38 -07:00
chenyuandGitHub ad24750487 const(value, dtype) -> const(value).cast(dtype) in tests (#17335) 2026-07-31 13:24:31 -04:00
chenyuandGitHub 7f4dbb8090 remove shape= from UOp.const [PR] (#17331) 
inlined to const_like
 2026-07-31 11:18:59 -04:00
chenyuandGitHub 8e2f175542 const(dtype, b) -> const(b, dtype) [PR] (#17328) 
prep for dtype removal
 2026-07-31 09:46:37 -04:00
chenyuandGitHub fe8ece7efa fix weak for image gate fusion [pr] (#17310) 2026-07-30 19:47:47 -04:00
chenyuandGitHub ce500c1946 broadcast doesn't cast CONST [pr] (#17308) 
* broadcast doesn't cast CONST [pr]

it might change arg from python int to python float, but not cast the CONST, so we don't need or_casted in symbolic. we lower the weak dtype right before decomp

* fix
 2026-07-30 18:51:59 -04:00
chenyuandGitHub b488cc7df2 update dtype_from_uop for SHR/SHL (#17307) 
certain backend cast the distance, and we should not re-broadcast it
 2026-07-30 13:57:58 -04:00
b1tgandGitHub e25bf77ce9 kimi delta attention (#17281) 
* kimi delta attention

* config

* test
 2026-07-30 08:23:56 -07:00
George HotzandGitHub b30c7e00d4 support 2d on UNSHARD (kimi) (#17285) 
* support 2d on UNSHARD

* fixes

* Fix test and spec

* single barrier

* 2d sharding works for devices too

* cleanups

* no _rewrap
 2026-07-29 12:01:59 -07:00
chenyuandGitHub dd86a30798 test case for weakfloat cast to weakint INDEX (#17287) 2026-07-29 14:15:52 -04:00
George HotzandGitHub 57ae1bc7a7 rename MULTI to UNSHARD (#17267) 
* rename MULTI to UNSHARD

* comment updates (glm)

* rename method to unshard
 2026-07-28 16:51:41 -07:00
George HotzandGitHub 1757067fa6 add device range as src[1] to multi (kimi) (#17264) 
* add device range as src[1] to multi (kimi)

* cleanup

* fix tests
 2026-07-28 16:39:34 -07:00
chenyuandGitHub 23e9e76e8c DEFAULT_FLOAT/DEFAULT_INT ContextVar [pr] (#17265) 2026-07-28 19:08:18 -04:00
chenyuandGitHub fde3a8faff CAPTURE_PROCESS_REPLAY=0 default chaging test [PR] (#17261) 
* CAPTURE_PROCESS_REPLAY=0 default chaging test [PR]

quick patch before making default dtype context vars, which is quite annoying

* few more
 2026-07-28 17:32:35 -04:00
qazalandGitHub e3b3eea1e2 test for extra copy in allreduce_cast (#16740) 
* test for extra copy in allreduce_cast

* simpler

* cleanup
 2026-07-28 19:02:54 +09:00
chenyuandGitHub 0bb36c9989 make Tensor(None) weakfloat (#17241) 
match other float consts
 2026-07-27 16:45:28 -04:00
chenyuandGitHub 818a892ebc flip from_py to use weak dtypes [pr] (#17229) 2026-07-27 10:07:00 -04:00
chenyuandGitHub 4b6760539b fix _prepare_jit_inputs for weak [pr] (#17222) 2026-07-26 14:03:27 -04:00
chenyuandGitHub a8d51097dc realize weak is no-op [pr] (#17219) 
None device and weak dtype are both virtual
 2026-07-26 11:52:56 -04:00
Manan GuptaGitHubGeorge Hotz
 		79c07a334c fix ValueError in UOp.axis for shard reshape crossing boundary (#16547) 
Co-authored-by: George Hotz <[email protected]>
 2026-07-26 08:38:10 -07:00
chenyuandGitHub 5d1aa84901 SHR/SHL are Broadcastable [pr] (#17216) 2026-07-26 02:11:03 -04:00
chenyuandGitHub a96974e70c slight weak behavior tweak and cleanups [pr] (#17214) 
* slight weak behavior tweak and cleanups [pr]

* ruff
 2026-07-25 22:51:24 -04:00
chenyuandGitHub 076b37e1ae failing batch norm test (#17206) 
* failing batch norm test

running stats does not schedule in training now since there's no reader

* not that
 2026-07-25 18:46:49 -04:00
chenyuandGitHub 74c2121d99 promo (uint64, int) -> weakfloat like JAX [pr] (#17204) 2026-07-25 15:20:42 -04:00
chenyuandGitHub 2dd2e3d786 don't support Tensor(list(np.array)) [pr] (#17175) 2026-07-24 03:13:28 -04:00
chenyuandGitHub 3e031d6c4d AND OR XOR SHL SHR cannot have float operands [PR] (#17164) 
* AND OR XOR SHL SHR cannot have float operands [PR]

* dont
 2026-07-23 18:51:56 -04:00
chenyuandGitHub 9267fca91a use STACK in wino (#17155) 
STACK is great
 2026-07-23 13:02:42 -04:00
chenyuandGitHub 2983987321 clean up some tests (#17154) 
duplicated or typo or slow
 2026-07-23 11:22:18 -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
chenyuandGitHub b1060ca708 don't promote dtype in _pad_constant [pr] (#17125) 2026-07-21 18:13:50 -04:00
chenyuandGitHub f19a2ad771 single where mixin [pr] (#17118) 
* single where mixin [pr]

no shape broadcasting in ufix and _broadcasted anymore

* QCOM vectorized bool is broken
 2026-07-21 17:38:36 -04: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 46b82d4755 don't auto cast cond for WHERE (#17115) 
no or_casted all WHEREs with single mixin, matched torch
 2026-07-21 13:00:07 -04:00
chenyuandGitHub 76dade5a11 implicit broadcast gradient based on shape only [pr] (#17114) 
fixed gradient for shape () UOp, enabled unify WHERE mixin
 2026-07-21 12:51:54 -04:00
chenyuandGitHub f64f96ec59 broadcast_axes [PR] (#17112) 
prerequisite to simplify broadcasting logic and make it implicit
 2026-07-21 11:53:12 -04:00
chenyuandGitHub 95f5c85bf3 some realize and corealize for slow tests (#17106) 2026-07-20 21:43:23 -04: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 47629f4bcf more weak dtype materialization raise (#17071) 2026-07-17 23:15:14 -04:00
chenyuandGitHub f315df29a0 no weak Tensor from and to real buffer (#17067) 
* no weak Tensor from and to real buffer

creation, assign, safe_save

* is_numpy_ndarray to tensor

* one more
 2026-07-17 16:09:10 -04:00
chenyuandGitHub 88826a6f35 no weak dtype for randn_like either (#17055) 2026-07-16 18:29:12 -04:00
chenyuandGitHub 3bfd62e915 fix 0 size tolist to match numpy (#17054) 2026-07-16 17:42:52 -04:00
chenyuandGitHub 6fa0b2b19e materialize weak dtype casts to default (#17051) 
in clone and _buffer
 2026-07-16 16:12:33 -04:00
chenyuandGitHub f6a92d0a16 sum_acc_dtype(weak) is weak (#17047) 
also no explicit weak for rand
 2026-07-16 14:32:37 -04:00
chenyuandGitHub 5a4156c5d1 bitcast and element_size raise for weak dtypes (#17046) 2026-07-16 13:07:45 -04:00
chenyuandGitHub dba8b6b505 allow weak alu operands (#17044) 2026-07-16 09:33:20 -04:00
chenyuandGitHub e8d3047a50 dtype_from_uop cleanup [PR] (#17036) 2026-07-15 21:52:21 -04:00
chenyuandGitHub c23d8188e1 remove _ensure_float [pr] (#17030) 
do this cast late. allow `SQRT(int)`
 2026-07-15 11:19:20 -04:00
chenyuandGitHub ebe26420a7 update where Invalid rules [pr] (#17026) 
fixed TestInvalidTensor.test_tensor_index
 2026-07-15 00:00:30 -04:00