chenyu and GitHub
086f45cb81
dtype is not a UOp field anymore ( #17821 )
2026-08-28 16:17:29 -04:00
chenyu and GitHub
bd2711b7c1
put dtype to Ops.INS arg ( #17817 )
2026-08-28 15:05:17 -04:00
George Hotz and GitHub
6b361e4de7
move shape into arg for param/buffer ( #17783 )
...
* move shape into arg for param/buffer
* no param_from_shape
* drop gratuitous syntax changes
* image is a in-graph view, folded into the param arg at render; drop dead multi param sharding
* view_as helper, simpler resolve_function, spec update
* spec: param/buffer are flat storage, no shape input
* image dims live in the param arg from transform_to_image; tighten kernel graph spec
* kernel graph spec: only RESHAPE/SHRINK over storage values, not all movement
* kernel graph: call args are storage, not views (pm_no_view_args); assert in spec
* strip views at the kernel graph level (pm_no_views), move into rangeify
* touchups
2026-08-27 18:13:24 -07:00
George Hotz and GitHub
f267a49639
UOp.param accepts single sint as shape (int,) ( #17784 )
...
* UOp.param accepts single sint as shape (int,)
Change UOp.param signature from shape:tuple[sint, ...]|None to shape:tuple[sint, ...]|sint|None.
A single sint means 1D shape (sint,). Update all callers from (n,) to n syntax.
No param_from_shape — multi-dim shapes stay as tuples.
* use single int syntax in hcq2 copy_with_kernel
2026-08-27 14:07:33 -07:00
chenyu and GitHub
4bdc865131
delete unused rewrite rules [PR] ( #17748 )
2026-08-26 10:50:42 -04:00
b1tg and GitHub
9860e5d285
llm tokenizer: fix tekken, add gpt4o ( #17733 )
2026-08-25 23:30:35 -07:00
George Hotz and GitHub
dc04c7820e
lil fixes from new_rangeify ( #17741 )
...
* lil fixes from new_rangeify
* gpt sol review
2026-08-25 17:15:28 -07:00
chenyu and GitHub
6ece327cf3
CUSTOM arg is (str, dtype) [PR] ( #17737 )
2026-08-25 20:15:06 -04:00
chenyu and GitHub
9f01775cf4
dtype_from_uop for THREEFRY and FDIV ( #17727 )
2026-08-25 08:43:59 -04:00
chenyu and GitHub
9607787ce1
delete more explicitly set dtype to UOp [PR] ( #17726 )
...
* delete more explicitly set dtype to UOp [PR]
* not that
2026-08-25 08:32:45 -04:00
George Hotz and GitHub
a5678317c2
split rangeify to prepare.py ( #17722 )
2026-08-24 19:12:13 -07:00
chenyu and GitHub
a7df1a1ace
const are weak 3 [pr] ( #17695 )
2026-08-24 20:32:57 -04:00
chenyu and GitHub
8b164aefea
test update from weak const ( #17689 )
2026-08-23 17:30:16 -04:00
chenyu and GitHub
356f665377
test update for weak const ( #17675 )
2026-08-21 21:50:29 -04:00
chenyu and GitHub
daa154aa22
FLOORDIV to SHR for powers of 2 [pr] ( #17669 )
2026-08-21 16:28:04 -04:00
chenyu and GitHub
8f9cbdf0cc
few more self folding [pr] ( #17657 )
2026-08-21 15:21:28 -04:00
chenyu and GitHub
554d078ac4
fix render_marg [pr] ( #17656 )
2026-08-21 10:03:22 -04:00
qazal and GitHub
1c3c9e96f6
remove setUp from test_sqtt_profiler ( #17652 )
...
* remove setUp from test_sqtt_profiler
* remove that
* cleaner
* do not need that
2026-08-21 17:29:11 +09:00
qazal and GitHub
e8a8d99b99
profiler tracing using unique keys ( #17651 )
...
* profiler tracing using unique keys
* cleaner + comment
2026-08-21 17:06:01 +09:00
qazal and GitHub
dcc2d021e7
prereq viz cleanups for unique profile keys ( #17649 )
...
* cleaner
* just use VIZ=-2
* better
2026-08-21 16:18:25 +09:00
qazal and GitHub
80bf60d782
profiler failing test for non unique kernel names ( #17647 )
...
* err
* expected
2026-08-21 13:52:35 +09:00
George Hotz and GitHub
3fdbb82bfe
add ansipad and PARALLEL contextvar ( #17642 )
2026-08-20 16:26:02 -07:00
chenyu and GitHub
592e3f8363
update const selector where folding [pr] ( #17640 )
...
folding a strong dtype WHERE to a weak const branch keeps the strong dtype
2026-08-20 18:49:34 -04:00
sirhcm and GitHub
3715006a21
fix float_to_bf16 on non-float32 inputs ( #17638 )
2026-08-20 18:14:30 -04:00
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
sirhcmandGitHub