chenyu and GitHub
e6324d1e1c
test updates from weak const branch ( #17618 )
2026-08-19 22:56:25 -04:00
sirhcm and GitHub
2067133732
cpu: link with rt ( #17608 )
2026-08-19 17:55:08 -04:00
chenyu and GitHub
fc214da417
test updates for weak const change ( #17606 )
2026-08-19 15:56:01 -04:00
chenyu and GitHub
b8cc74ecf8
no float in tensor shape [pr] ( #17605 )
2026-08-19 15:40:16 -04:00
chenyu and GitHub
c218b4842d
fold_bitcast should truncate its input [pr] ( #17601 )
2026-08-19 14:39:36 -04:00
chenyu and GitHub
bd6e70ac15
delete stale tests ( #17596 )
2026-08-19 11:18:05 -04:00
chenyu and GitHub
9550378704
finish casted_consts migration [PR] ( #17595 )
2026-08-19 10:43:55 -04:00
chenyu and GitHub
b3e2f17b24
update NULL tests that depends on strong dtype CONST ( #17594 )
2026-08-19 10:26:38 -04:00
chenyu and GitHub
68b4407fe3
casted CONST migration for cstyle [pr] ( #17587 )
2026-08-19 09:01:40 -04:00
George Hotz and GitHub
c31038ff37
use KernelCountException when kernel count is being compared ( #17584 )
2026-08-18 16:06:03 -07:00
chenyu and GitHub
34c9b9d434
add back cast where rule [pr] ( #17572 )
2026-08-17 15:27:20 -04:00
chenyu and GitHub
138fb4a783
delete dead DType.scalar [PR] ( #17561 )
2026-08-16 21:12:17 -04:00
chenyu and GitHub
26cbadd69a
no pm_fold_cast_const in full_rewrite_to_sink [PR] ( #17547 )
2026-08-15 10:18:59 -04:00
chenyu and GitHub
4b0525e594
no pm_fold_cast_const in UOp.simplify and hcq2 [pr] ( #17540 )
2026-08-14 21:39:47 -04:00
chenyu and GitHub
13c381b0c0
remove pm_fold_cast_const from initial symbolic [pr] ( #17535 )
...
interestingly it gives more accurate numerics when composing const like log10
2026-08-14 14:14:40 -04:00
chenyu and GitHub
80169c6758
remove where push cast to branches from sym [pr] ( #17533 )
...
* remove where push cast to branches from sym [pr]
not really needed and one less place that generates casted weak const when it's not needed
* fix
2026-08-14 12:57:56 -04:00
George Hotz and GitHub
b1859805b1
remove Ops.BIND ( #17511 )
...
* remove Ops.BIND
* param arg
* simplify that
* simplify
* cleaner
* props, not functions
* param and buffer can share
2026-08-13 23:52:16 -07:00
Raine and GitHub
de04781b36
simplify equivalent const max ( #17505 )
...
* add const max folds
* add regression test
* move
2026-08-12 08:39:29 -07:00
sirhcm and GitHub
479ffb0cda
remove Ops.SLICE ( #17492 )
2026-08-11 18:50:04 -04:00
George Hotz and GitHub
d4d537c8ae
add SPEC checking for the kernel graph ( #17432 )
...
* add SPEC checking for the kernel graph
* skip test
* raise there, not None
* handwritten
* issue with unshard
* multi works
* and bitcast
* fix new tests
* fix linear
* remove call index
* fix shrink
* fixes
2026-08-08 10:00:06 -07:00
chenyu and GitHub
4a3b8f6501
better _drop_valid_stmts [pr] ( #17454 )
2026-08-07 15:18:35 -04:00
chenyu and GitHub
f76422b8af
fix cast to float _min_max [pr] ( #17451 )
2026-08-07 11:56:19 -04:00
chenyu and GitHub
9020a88f03
truncate float in DType.const [pr] ( #17439 )
2026-08-06 20:19:08 -04:00
chenyu and GitHub
28e6ef6937
fix postopt symbolic [pr] ( #17433 )
...
REDUCE with src simplied to const would become unparented
2026-08-06 00:12:30 -04:00
chenyu and GitHub
7a9cd8e329
move weak function and pm to uop/weak [PR] ( #17429 )
2026-08-05 22:27:40 -04:00
chenyu and GitHub
d726e5f7f3
split pm_fold_cast_const [PR] ( #17425 )
...
need to delete this rule that writes strong typed CONST
2026-08-05 19:56:04 -04:00
George Hotz and GitHub
581bfdd94f
merge track_rewrites and profile_matches into rewrite_group [PR] ( #17420 )
...
* merge track_rewrites and profile_matches into rewrite_group
* bug
* flip ctx polarity
2026-08-05 14:41:38 -07:00
George Hotz and GitHub
46f0003776
more KernelCountException ( #17407 )
2026-08-04 22:55:38 -07:00
chenyu and GitHub
9b508dfafc
remove invalid special case in cast [PR] ( #17405 )
2026-08-04 23:08:07 -04:00
George Hotz and GitHub
e1f42681fa
add new schedule tests + format better ( #17402 )
...
* add new schedule tests + format better
* assert_kernel_count
2026-08-04 18:46:38 -07:00
George Hotz and GitHub
6122b3c98f
use check_schedule in tests where possible ( #17400 )
2026-08-04 18:17:29 -07:00
chenyu and GitHub
c1a10e0726
fix _min_max for CAST from float to int [pr] ( #17396 )
...
* fix _min_max for CAST from float to int [pr]
* fix
2026-08-04 18:06:50 -04:00
George Hotz and GitHub
c2625c78cb
scalar ALU index fix + llm: preserve_thinking ( #17381 )
...
* cstyle: scalar ALU index fix, serve: preserve_thinking, test: fix Handler import
- cstyle.py: return scalar directly when ALU buffer has 1 element
- cli.py: add preserve_thinking param to FallbackTemplate.render
- serve.py: pass preserve_thinking=True when rendering chat completions
- test_llm_server.py: fix import to use Handler from llm.serve
* real fix
2026-08-03 15:10:21 -07:00
George Hotz and GitHub
05bc7c6994
fix llm reasoning and Linear import ( #17372 )
2026-08-02 12:31:52 -07:00
chenyu and GitHub
59df317b12
use UOp.const to create new consts [PR] ( #17368 )
...
replace arg won't work with ConstArg
2026-08-02 13:31:55 -04:00
chenyu and GitHub
8e524ca467
CONST related cleanups [pr] ( #17352 )
...
ConstFloat(nan) != nan should be False, and some Invalid bool cleanups
2026-08-01 09:54:28 -04:00
chenyu and GitHub
6c0ec39279
UOp.is_invalid [PR] ( #17351 )
...
helper to prep ConstArg
2026-08-01 02:36:32 -04:00
chenyu and GitHub
b502fc1367
more const arg -> val ( #17350 )
2026-08-01 02:18:10 -04:00
chenyu and GitHub
850989115d
__int__ and __float__ work for weak ( #17342 )
2026-07-31 19:56:37 -04:00
chenyu and GitHub
277433259e
fix sym_infer for CAST ( #17338 )
2026-07-31 14:38:38 -04:00
chenyu and GitHub
ad24750487
const(value, dtype) -> const(value).cast(dtype) in tests ( #17335 )
2026-07-31 13:24:31 -04:00
chenyu and GitHub
7f4dbb8090
remove shape= from UOp.const [PR] ( #17331 )
...
inlined to const_like
2026-07-31 11:18:59 -04:00
chenyu and GitHub
8e2f175542
const(dtype, b) -> const(b, dtype) [PR] ( #17328 )
...
prep for dtype removal
2026-07-31 09:46:37 -04:00
nimlgen and GitHub
0c4bfaeb48
coalesce ints ( #17323 )
...
* merge ints
* fix z3 validation of coalesced loads
* fix uint vector names in CUDA and Metal
2026-07-31 15:49:56 +03:00
chenyu and GitHub
fe8ece7efa
fix weak for image gate fusion [pr] ( #17310 )
2026-07-30 19:47:47 -04:00
chenyu and GitHub
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
chenyu and GitHub
d52ef3077c
fix do_devectorize dtype ( #17295 )
...
each one derives its dtype from src, not shared
2026-07-29 23:04:25 -04:00
chenyu and GitHub
fd912b348c
generic c0*x<c1 [pr] ( #17291 )
2026-07-29 18:45:23 -04:00
chenyu and GitHub
027907a544
fix c0*x<c1 symbolic [pr] ( #17290 )
...
use `//` and not math.ceil
2026-07-29 18:32:31 -04:00
George Hotz and GitHub
52c9e5a99e
rename LOOP -> WEAK and STRONGLOOP -> LOOP ( #17283 )
2026-07-29 10:38:36 -07:00