George Hotz and GitHub
1a523193dd
Merge branch 'master' into invalid_try_3
2026-07-25 14:07:16 -07:00
geohot
e70a79465e
invalid try 3 (kimi)
2026-07-25 14:06:29 -07:00
chenyu and GitHub
492dc6d5fb
delete _index_to_concrete_int [pr] ( #17205 )
...
staying weak is okay
2026-07-25 16:05:08 -04:00
chenyu and GitHub
74c2121d99
promo (uint64, int) -> weakfloat like JAX [pr] ( #17204 )
2026-07-25 15:20:42 -04:00
chenyu and GitHub
f902513355
derive torch backend dispatch from the aten schema ( #17203 )
2026-07-25 14:21:38 -04:00
chenyu and GitHub
ee2ccb1f24
put weakint in dtypes.weaks [pr] ( #17202 )
...
* put weakint in dtypes.weaks [pr]
* custom_add_var
2026-07-25 14:00:34 -04:00
wozeparrot and GitHub
f0117e98df
refactor mlperf optim ( #17200 )
2026-07-25 10:37:11 -07:00
chenyu and GitHub
983ad3bd95
fix torch backend batchnorm backward ( #17201 )
2026-07-25 13:15:09 -04:00
chenyu and GitHub
9f78504304
checked cast in torch backend unwrap ( #17199 )
2026-07-25 12:34:08 -04:00
qazal and GitHub
732e6bd52c
add one line viz mention ( #17198 )
...
* add one line viz mention
* changes
* edit
2026-07-26 00:18:41 +09:00
chenyu and GitHub
8a10892f5a
fix torch backend as_strided ( #17195 )
...
0 means 0 offset
2026-07-25 02:24:17 -04:00
chenyu and GitHub
4c58b260fb
less wrong calculate_storage_offset ( #17194 )
...
initially for speed, then realized it's just wrong
2026-07-25 01:53:54 -04:00
chenyu and GitHub
d923263a5b
simpler lower_weakint_node [PR] ( #17193 )
...
works since weakint is in promo lattice properly now
2026-07-25 00:31:04 -04:00
sirhcm and GitHub
9fdaa4bff2
standardize Program class ( #17189 )
2026-07-25 00:17:38 -04:00
chenyu and GitHub
0a6125e571
Invalid is bool, put weakint in promo lattice [pr] ( #17188 )
2026-07-25 00:06:53 -04:00
chenyu and GitHub
de3508e375
update create_non_native_float_pats [PR] ( #17191 )
...
match based on WHERE dtype, not branch dtyle to prep for new Invalid
2026-07-24 21:21:16 -04:00
George Hotz and GitHub
a346e2ea7b
improve call for viz ( #17187 )
2026-07-24 15:26:05 -07:00
George Hotz and GitHub
f65001e297
cleanups because copy is not allowed in rangeify (kimi) ( #17186 )
...
* cleanups because copy is not allowed in rangeify (kimi)
* gpt cleanups
2026-07-24 14:17:10 -07:00
George Hotz and GitHub
dcad119415
convert COPY -> STORE early ( #17172 )
...
* convert COPY -> STORE early
* finish it (glm)
* no slop
* fixes
* not None
* copy one
* bugs fixed, just kernel count i think
* fix disk to disk
* move out of realize
* revert that
* simpler
* fixes
* add copy kernel simplification
* fixes
* dead copy
* remove reshape on store
* fix mstack lowering
2026-07-24 13:49:17 -07:00
chenyu and GitHub
38c04fa648
data_src concept in run_rangeify [pr] ( #17183 )
...
fixed a skip based on `dtypes.weakint`, it was pulling shape args
2026-07-24 14:20:41 -04:00
chenyu and GitHub
86b42d7cb8
clean up unbroadcast [PR] ( #17182 )
...
also renamed
unbroadcast -> pm_expand_broadcast,
broadcast_binary -> expand_broadcast
2026-07-24 11:48:07 -04:00
chenyu and GitHub
e0b30b7a1d
simpler _split_cumalu [PR] ( #17180 )
2026-07-24 10:52:50 -04:00
Raine and GitHub
fcb1d3fc9f
add x86 loops ( #17179 )
...
* add x86 loops
* lint
2026-07-24 07:26:09 -07:00
wozeparrot and GitHub
c9b60caf8c
gptoss: moe gemm kernels ( #17178 )
2026-07-24 07:25:15 -07:00
chenyu and GitHub
40bfd84956
remove unneeded expand [PR] ( #17176 )
2026-07-24 10:03:46 -04:00
nimlgen and GitHub
017d1ab56e
cpu prep for hcq2 ( #17153 )
...
* cpu prep for hcq2
* x
2026-07-24 15:57:21 +03:00
nimlgen and GitHub
0ecef210bb
hcq2 cleanup 2 ( #17177 )
...
* hcq uops
* x
2026-07-24 13:04:21 +03:00
nimlgen and GitHub
709d7b2343
hcq2 cleanups ( #17162 )
2026-07-24 10:14:09 +03:00
chenyu and GitHub
2dd2e3d786
don't support Tensor(list(np.array)) [pr] ( #17175 )
2026-07-24 03:13:28 -04:00
George Hotz and GitHub
08eceb0ad9
llm: fix generated tokens in usage accounting ( #17174 )
2026-07-23 21:38:07 -07:00
chenyu and GitHub
a292dab8ab
corner cases from weak const branch ( #17173 )
2026-07-24 00:09:56 -04:00
chenyu and GitHub
f5d9c31d15
arange upcast to int64 with big N ( #17167 )
...
simplifies _one_hot_along_dim
2026-07-23 20:11:39 -04:00
George Hotz and GitHub
bf5989ea48
support loops in nir (kimi) ( #17166 )
...
* support loops in nir
* tpying
2026-07-23 17:05:15 -07:00
chenyu and GitHub
0f9edb02f6
don't use promo_lattice in fast_idiv [PR] ( #17165 )
...
int64 will link to weakfloat
2026-07-23 19:20:53 -04:00
chenyu and GitHub
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
chenyu and GitHub
1b616669d2
few more symbolic rules [pr] ( #17161 )
...
bitcast twice can skip middle one; bool != False is self; two more where combine
2026-07-23 18:05:32 -04:00
sirhcm and GitHub
6ee16eb1fa
split out LLVM compilers ( #17160 )
2026-07-23 17:45:23 -04:00
chenyu and GitHub
00cf9c0c25
tweak add CONTIGUOUS to tagged UOps [PR] ( #17159 )
...
dont call replace when the tag is alreayd None
2026-07-23 17:06:56 -04:00
chenyu and GitHub
528aa4fc8c
more stack const in gguf ( #17157 )
...
better IQ4_XS and IQ3_XXS
2026-07-23 14:04:04 -04:00
chenyu and GitHub
250de4b142
use stack for same shape cat [pr] ( #17156 )
...
same device multi cat benefits too
2026-07-23 13:50:28 -04:00
chenyu and GitHub
9267fca91a
use STACK in wino ( #17155 )
...
STACK is great
2026-07-23 13:02:42 -04:00
chenyu and GitHub
2983987321
clean up some tests ( #17154 )
...
duplicated or typo or slow
2026-07-23 11:22:18 -04:00
chenyu and GitHub
33279b5d4c
guard move add/mul consts to end [PR] ( #17152 )
...
make sure the rules won't cycle
2026-07-23 10:53:05 -04:00
wozeparrot and GitHub
2e7db75668
gptoss: faster swa ( #17150 )
2026-07-23 06:56:27 -07:00
nimlgen and GitHub
62273d50fc
simpler hcq buffer ( #17148 )
...
* simpleer hcq buffer
* x
2026-07-23 13:28:51 +03:00
chenyu and GitHub
c5b2b9242d
python speed tweak for compile3 ( #17147 )
2026-07-23 02:02:44 -04:00
chenyu and GitHub
e9a86c99ed
more symmetrical x//d<c -> x<c*d [pr] ( #17146 )
...
* more symmetrical x//d<c -> x<c*d [pr]
* parse_valid too
2026-07-23 00:19:53 -04:00
chenyu and GitHub
6a9c23b1c1
more symmetrical div_and_mod_symbolic [pr] ( #17145 )
2026-07-22 23:34:03 -04:00
chenyu and GitHub
723309b5c0
update hand_coded_optimizations [pr] ( #17144 )
...
only upcast when they share valids
2026-07-22 22:26:27 -04:00
chenyu and GitHub
2aafca411d
Revert "reorder sym in full_rewrite_to_sink [pr] ( #17141 )" ( #17142 )
...
This reverts commit 7755480f46 .
2026-07-22 19:47:47 -04:00