Commit graph

13,698 commits

Author SHA1 Message Date
George Hotz
30cc426916 order 2026-06-23 12:39:06 -07:00
George Hotz
c7fd74d523 order 2026-06-23 12:30:19 -07:00
George Hotz
4e398e3f1f lengths 2026-06-23 09:18:51 -07:00
George Hotz
5238f304c7 coalese 2026-06-23 09:04:23 -07:00
George Hotz
87e12fad81 cleanup 2026-06-23 08:43:00 -07:00
George Hotz
5eaf02c719 memory coalesing 2026-06-23 08:30:23 -07:00
George Hotz
4424af7bd8 revert that 2026-06-22 18:36:50 -07:00
George Hotz
c01d75a651 fixes 2026-06-22 18:18:19 -07:00
George Hotz
63ec8ad21d
Merge branch 'master' into codegen_try_2 2026-06-22 17:44:37 -07:00
George Hotz
c9dc1d63cc
small changes from new codegen (#16712)
* small changes from new codegen

* shrink/flatten
2026-06-22 17:44:15 -07:00
Christopher Milan
da98fae9e1
ci: try parallelizing tc tests (#16710) 2026-06-22 20:43:32 -04:00
chenyu
15988b5941
contiguous to mixin and cleanups [PR] (#16711) 2026-06-22 20:18:18 -04:00
George Hotz
cccd9c2c03 loads are grouped 2026-06-22 16:12:18 -07:00
Christopher Milan
cbfcf36e44
ci: remove generate_dataset and CL misc (#16709) 2026-06-22 18:01:07 -04:00
nimlgen
f9c8c697d6
hcq2: drop args after inner deps (#16708) 2026-06-22 23:26:11 +03:00
chenyu
0138480910
dropout and scaled_dot_product_attention to mixin (#16707) 2026-06-22 16:17:45 -04:00
George Hotz
303b6ba14c
Merge branch 'master' into codegen_try_2 2026-06-22 13:02:37 -07:00
George Hotz
2bf3c48c1b reduce hack remove 2026-06-22 13:01:36 -07:00
chenyu
33b635d23a
Tensor.train -> TRAINING [PR] (#16705)
* Tensor.train -> TRAINING [PR]

* doc
2026-06-22 15:13:22 -04:00
George Hotz
ba75b68c12 simple 2026-06-22 11:36:13 -07:00
George Hotz
958088cc13 Merge remote-tracking branch 'origin/master' into codegen_try_2 2026-06-22 11:25:46 -07:00
chenyu
625d8bbd0d
TRAINING ContextVar (#16703) 2026-06-22 13:03:08 -04:00
George Hotz
257ed03f57 fix store 2026-06-22 08:46:56 -07:00
George Hotz
15476572cd
Merge branch 'master' into codegen_try_2 2026-06-22 08:34:49 -07:00
wozeparrot
fe9b19b12d
llama: more mp mem fixes (#16701)
* llama: more mp mem fixes

* clean: unused

* fix: batch
2026-06-22 10:54:35 -04:00
chenyu
267af9c601
full_like to CreationMixin [PR] (#16702) 2026-06-22 09:33:23 -04:00
chenyu
97da54b9d6
more method to CreationMixin [PR] (#16698) 2026-06-22 00:01:22 -04:00
chenyu
fd0dc40689
clean up CreationMixin and DTypeMixin [PR] (#16697) 2026-06-21 21:13:40 -04:00
chenyu
2d8b802958
contiguous in wino conv (#16696)
also fixed test_counters
2026-06-21 17:11:46 -04:00
chenyu
ba1d3baae8
masked_select and nonzero to mixin [PR] (#16695)
with a .data stub
2026-06-21 15:10:44 -04:00
chenyu
d80a41d559
some rand method to RandMixin [PR] (#16693) 2026-06-21 12:16:51 -04:00
wozeparrot
5164c21b44
gemm: keep shape thru mxfp8 quantize (#16692) 2026-06-20 22:28:53 -07:00
chenyu
58ff75272e
const_like and invalids to mixin [PR] (#16690)
* const_like and invalids to mixin [PR]

* empty_like

* einsum

* type
2026-06-21 00:02:29 -04:00
chenyu
b50da5c205
move Tensor.__getitem__ to mixin [PR] (#16689) 2026-06-20 22:01:45 -04:00
chenyu
4618d27129
final const cleanups [PR] (#16688) 2026-06-20 21:38:16 -04:00
chenyu
9ae0a93d0e
more const cleanups [PR] (#16682) 2026-06-20 20:41:43 -04:00
George Hotz
f8949a0de1 fix exec_alu 2026-06-20 17:01:28 -07:00
George Hotz
62c6c75657 test tiny almost passes 2026-06-20 16:46:01 -07:00
George Hotz
530aed739d devec 2026-06-20 08:34:20 -07:00
George Hotz
da402953d9 new codegen, try 2 2026-06-20 08:30:38 -07:00
George Hotz
30830850a9
small changes from new codegen (#16681)
* small changes from new codegen

* revert that
2026-06-19 18:29:01 -07:00
chenyu
8b07cca9f7
invalid clone try 3+ [PR] (#16679) 2026-06-19 20:13:52 -04:00
Christopher Milan
b2199c54a3
ci: update actions/cache/restore to suppress warnings (#16680) 2026-06-19 18:27:52 -04:00
Christopher Milan
1822eed8d3
ci: only test models on cpu (#16678) 2026-06-19 18:16:59 -04:00
wozeparrot
bba611bb59
gemm: fix mxfp8 on more shapes (#16677) 2026-06-19 13:28:53 -07:00
chenyu
67c3e589a1
invalid clone tests and prereq [PR] (#16675) 2026-06-19 13:20:43 -04:00
George Hotz
649971f02a
remove DEFINE_LOCAL and DEFINE_REG (gpt) (#16673)
* remove define_local and define_reg (gpt)

* fix precommit

* cleanups

* regalloc fix

* cleanups 2
2026-06-19 10:07:50 -07:00
George Hotz
b05bea81ce
x86 cleanups (fable) [pr] (#16591)
* x86 cleanups (fable)

* support shrink

* remove ptr dtype

* move that

* is_lane helper

* Revert "is_lane helper"

This reverts commit ea4571254d.
2026-06-19 09:04:51 -07:00
nimlgen
97c2e7a3d9
spec: add getaddr (#16674) 2026-06-19 15:37:33 +03:00
George Hotz
d7b10c69bc
update placeholder to not create DEFINE_LOCAL/DEFINE_REG (#16671)
* update placeholder to not create DEFINE_LOCAL/DEFINE_REG

* simpler

* define_local
2026-06-18 21:21:06 -07:00