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
George Hotz and GitHub
d65ea465ed
cleanup gemm fragment + add store unshard ( #17313 )
...
* cleanup gemm fragment + add store unshard
* multi
* fix
2026-07-30 20:53:28 -07:00
qazal and GitHub
7b7b9c1ae0
spec tests for movement ops before custom_kernel ( #17299 )
...
* spec
* fixing
* xfail
* remove
* update tests
* add back
* change to max_kernels
* split and cleanup
2026-07-30 19:20:24 +09:00
George Hotz and GitHub
138676ab81
improve fragment example + index unshard (kimi) ( #17288 )
...
* fix dtypes in fragment example
* match tilelang
* flip locals
* fix index on unshard
* test fixes
* kimi needs more taste
2026-07-29 15:38:38 -07:00
qazal and GitHub
6ea7d366fa
test permuted input in custom Ops.PROGRAM test ( #17279 )
...
* test permuted input in custom Ops.PROGRAM test
* dtype
2026-07-29 18:29:05 +09:00
George Hotz and GitHub
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
drk and GitHub
d1c3ae0f91
preserve typed ranges through reshape ( #17259 )
2026-07-28 16:51:06 -04:00
qazal and GitHub
95e3b0066f
webgpu failing test for duplicate PARAM in CALL [pr] ( #17230 )
...
* webgpu failing test for duplicate PARAM in CALL [pr]
* typo
2026-07-27 14:06:06 +09: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
6b82c0cb95
minor test fixes ( #17138 )
...
TestCustomKernel square().max() masks NaN, and fixed test_tensor_cores_multi_reduce
2026-07-22 15:08:52 -04:00
George Hotz and GitHub
fdffc6c0c8
remove dtypes base ( #16931 )
...
* remove dtypes base
* find/replace bug
* vcount is junk too
2026-07-08 11:48:11 -07:00
George Hotz and GitHub
a145fdce4b
remove the .gep method ( #16867 )
2026-07-04 14:07:09 -07:00
chenyu and GitHub
3a9b87e48f
remove Ops.DEVICE from Ops.PROGRAM [PR] ( #16797 )
2026-06-29 17:41:31 -04:00
chenyu and GitHub
1e08c0a07c
remove NOOP from AFTER with multiple srcs ( #16646 )
2026-06-17 14:35:02 -04:00
chenyu and GitHub
1acc40600d
indexing an after with all fully invalid stores is invalid ( #16643 )
...
* indexing an after with all fully invalid stores is invalid
* typing cast
2026-06-17 11:06:36 -04:00
qazal and GitHub
4d34590b7d
llama: less E kernels ( #16517 )
2026-06-12 19:49:25 +09:00
qazal and GitHub
34481830f1
rangeify: fix cost function for AFTER(out, CALL) ( #16559 )
...
* simple failing test
* fix rangeify cost function
* new ops count
2026-06-10 17:30:50 +09:00
chenyu and GitHub
fa14cde05c
test update for arange and eye ( #16394 )
...
these will need explicit clone to make a buffer
2026-05-26 22:48:34 -04:00
chenyu and GitHub
8631b6f17d
remove use of requires_grad in test/ ( #16237 )
2026-05-16 17:21:07 -04:00
wozeparrot and GitHub
2d48d7ab09
remove more invalid ( #16227 )
2026-05-16 02:52:27 -07:00
George Hotz and GitHub
64c81dfd24
add all codegen stages to spec_tensor ( #16163 )
2026-05-12 10:35:38 -07:00
qazal and GitHub
fe39cf148a
add Ops.SOURCE test ( #16155 )
...
* simple failing test
* raises
* change
2026-05-12 22:49:32 +09:00
George Hotz and GitHub
8294d105a7
Update the spec in spec.py to match the current state ( #16132 )
...
* start work on specv2
* more spec
* more spec
* fix amd emulator
* more spec
* more
* fix test_uop_graph
* move those
* spec=2
* skip those questionable tests
* ptx fix
* more spec=2
* store
* allow custom function in tensor
* spec 2
* fix beam search for tensor cores
* delete the old specs
* fix import
2026-05-11 20:07:47 -07:00
qazal and GitHub
8b147a9ed5
minimal repro for llama copies 2 ( #16011 )
2026-05-01 22:23:47 +09:00
qazal and GitHub
58b34e71bd
failing test for llama useless copies ( #15989 )
2026-05-01 00:55:29 +09:00
nimlgen and GitHub
d2ab6ea7a6
remove schedule batch 3 ( #15924 )
...
* remove shcedule batch 3
* batch 6
* batch 7
2026-04-25 11:53:16 +03:00
chenyu and GitHub
7a1adfd2aa
update Tensor.allclose to return Tensor ( #15904 )
...
matches jax
2026-04-24 08:27:17 -04:00
chenyu and GitHub
9192c93b7e
Tensor.invalid -> Tesnor.invalids ( #15849 )
...
matches ones and zeros, and to not share name with UOp.invalid
2026-04-21 11:19:51 -04:00
chenyu and GitHub
8bd4fead26
UOp.size -> prod(max_shape) ( #15755 )
...
and more test updates
2026-04-15 22:41:30 -04:00
chenyu and GitHub
10c262ced8
update tests that use UOp.size ( #15753 )
2026-04-15 21:58:27 -04:00
chenyu and GitHub
8fdef2d3e4
mean/std/var to mixin ( #15593 )
2026-04-03 10:42:41 -04:00
chenyu and GitHub
fc5b94b902
fix UOp.where(const, const) ( #15560 )
...
* fix UOp.where(const, const)
* fix
2026-04-01 05:28:49 -04:00
chenyu and GitHub
f0eaac4235
reduce mixin ( #15523 )
2026-03-30 05:23:58 -04:00
George Hotz and GitHub
6109117af1
anonymous buffers are Invalid ( #15336 )
...
* anonymous buffers are Invalid
* unique_const
* work
* remove invalid writes
* test_anonymous_buffers_in_function
2026-03-18 14:52:56 +08:00
George Hotz and GitHub
677145b393
all consts have shapes ( #14959 )
...
* all consts have shapes
* vconst has shape too
* use normal schedule
* cast ptrdtype
* image
* bitcast issue + hack
2026-02-23 10:26:50 +08:00
George Hotz and GitHub
2611907afb
start ripping out old scheduler -- no maps ( #14909 )
...
* start ripping out old scheduler -- no maps
* no more metadata
2026-02-20 21:05:04 +08:00
George Hotz and GitHub
c331798201
move tests to test/backend ( #14691 )
...
* move tests to test/backend
* fix imports
* fix CI
* revert that one
* Fix formatting in README for test command
2026-02-12 11:09:44 +08:00