 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 |
|
|
|
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 |
|
|
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 |
|
|
nimlgenandGitHub
|
4257939e50
|
remove copyin/copyout from Buffer (#17020)
* remove copyin/copyout from Buffer
* x
* x
* x
* x
|
2026-07-14 19:47:22 +03:00 |
|
|
chenyuandGitHub
|
82fbca43c5
|
fix Tensor(np) dtype and support fp8 safetensor (#17019)
|
2026-07-14 09:31:21 -04:00 |
|
|
George HotzandGitHub
|
dde2e736e5
|
fix disable_gc decorator reentrancy (#16999)
|
2026-07-12 15:32:54 -07:00 |
|
|
chenyuandGitHub
|
df50e0814c
|
explicit error for unbound Variable in program (#16971)
also allow Tensor(UOp, dtype)
|
2026-07-10 16:23:59 -04:00 |
|
 sirhcmandGitHub
|
88c6f02abe
|
remove late_buffer_view, try 2 (#16963)
|
2026-07-10 00:46:26 -04:00 |
|
|
chenyuandGitHub
|
ba2c68b1ed
|
start dtypes.index [PR] (#16949)
|
2026-07-09 15:12:23 -04:00 |
|
|
chenyuandGitHub
|
c43a3fdebb
|
bitcast to mixin [PR] (#16924)
|
2026-07-07 20:03:29 -04:00 |
|
|
chenyuandGitHub
|
682b098542
|
fix multi write after read in create_schedule (#16831)
|
2026-07-02 15:12:48 -04:00 |
|
|
chenyuandGitHub
|
3e73a2542b
|
_apply_map_to_tensors always use walk rewrite (#16827)
with assign it's wrong to recursively apply, fixed a random bug that counter does not update after round 2
|
2026-07-02 10:43:25 -04:00 |
|
|
chenyuandGitHub
|
2e62dd308d
|
move fs_load and fs_store to nn (#16821)
similar to safe_load and safe_save
|
2026-07-01 22:01:15 -04:00 |
|
|
chenyuandGitHub
|
c8aed121cf
|
fix WAR in create_schedule (#16823)
* fix WAR in create_schedule
combined dep building, fixed a case that realize(x, y) != realize(y, x)
* fix
|
2026-07-01 22:00:51 -04:00 |
|
|
chenyuandGitHub
|
efd256b2a3
|
fix precompiled STORE+AFTER (#16822)
|
2026-07-01 17:07:37 -04:00 |
|