Nino Risteski
5eb1fd5d3c
cleanup: untrack wait Metal buffers ( #15954 )
2026-04-28 12:54:59 -07:00
nimlgen
77965a22e5
local optimize as rewrite ( #15953 )
...
* local optimize as rewrite
* better
* x
* slighly rename
* fix
* ugh
* remove
* x
* remove
* not weak
2026-04-28 22:51:04 +03:00
qazal
54f00e1013
sqtt: correct rdna4 structs ( #15948 )
2026-04-28 07:35:50 +09:00
qazal
c58fd85a99
sqtt: add needs_rocprof decorator ( #15947 )
...
* sqtt: add needs_rocprof decorator
* version string
2026-04-28 06:22:50 +09:00
chenyu
77f9125c21
move Tensor.pad to OpMixin ( #15946 )
2026-04-27 16:56:04 -04:00
nimlgen
4164666c72
programinfo ( #15942 )
...
* programinfo
* fix
* m
* x
* x
* changes
* x
* fix
* rm
2026-04-27 23:12:03 +03:00
chenyu
fe38d6de94
_pad_circular and _pad_reflect_replicate to mixin ( #15944 )
2026-04-27 16:07:05 -04:00
qazal
8c174bdad4
viz/sqtt: correct exec pipes ( #15885 )
...
* wmma
* p2
* test
* left
* work
* pickle
* handwritten failing tests
* start work
* test the pipes
* empirical evidence
* update rdna4 enum types
* VALU pipe 1
* TRANSCENDENTAL pipe
* transcendental function units
* reorder
* wmma pipe
* cleanup and notes
* smaller
* work
* diff cleanup
* pickle
* use se:1
* int
2026-04-28 05:05:49 +09:00
nimlgen
96165ff0d1
validate_with_cpu as rewrite ( #15938 )
...
* validate_with_cpu as rewrite
* compil
* x
* linter
* moved
* fix
2026-04-26 19:58:53 +03:00
nimlgen
117e9e22dd
estimates from graph ( #15937 )
...
* estimates from graph
* test
* x
2026-04-26 18:22:53 +03:00
nimlgen
bb652352c7
remove execitem ( #15932 )
...
* remove execitem
* f
* x
2026-04-25 19:33:04 +03:00
nimlgen
e0ff6cc15c
remove old schedule ( #15930 )
...
* remove old schedule
* tests
* r
* x
2026-04-25 16:46:36 +03:00
qazal
9a23de7d27
viz/cli: unify profile and rewrites, -s ALL default ( #15931 )
...
* work
* workg
* better
* cleanup
* better defaults
* --ls
* better
* work
* update llama
* update
2026-04-25 22:31:24 +09:00
nimlgen
a5e9ea7a60
remove schedule batch 4 ( #15927 )
...
* remove schedule batch 4
* fini
2026-04-25 12:36:55 +03:00
nimlgen
d2ab6ea7a6
remove schedule batch 3 ( #15924 )
...
* remove shcedule batch 3
* batch 6
* batch 7
2026-04-25 11:53:16 +03:00
nimlgen
3c8a2db870
remove schedule() from tests batch 2 ( #15923 )
...
* remove schedule() from tests batch 2
* batch 4
2026-04-25 10:44:41 +03:00
Denys Melnyk
1fdcb13bfb
webgpu: fix weight lookup in export_model after compile_net key change ( #15919 )
...
* fix lookup site in export_model_webgpu after refactoring
webgpu (sd): fix export_model weight lookup after compile_net changes
fix lookup site in export_model_webgpu after refactoring
* add regression test
2026-04-25 10:04:55 +03:00
Christopher Milan
57fbaa3d49
amd: fallback to llvm when comgr is not available ( #15914 )
2026-04-24 23:30:16 -04:00
nimlgen
d3378010ee
schedule() -> schedule_linear() in tests (batch 1) ( #15915 )
...
* schedule_with_vars -> linear_with_vars in tests
* tests batch 1
* batch 2
* estimate_uop
* simpler
* rm
2026-04-24 23:40:53 +03:00
chenyu
b501ba3e42
nll_loss to mixin ( #15918 )
2026-04-24 15:50:31 -04:00
chenyu
2f9fdb4a37
scatter to mixin ( #15917 )
2026-04-24 15:37:37 -04:00
nimlgen
f2751955cb
remove linear_to_schedule from tests ( #15912 )
...
* remove linear_to_schedule from tests
* x
2026-04-24 20:02:10 +03:00
chenyu
03a7604f76
sort argsort topk allclose to mixin ( #15910 )
2026-04-24 10:20:46 -04:00
chenyu
7a1adfd2aa
update Tensor.allclose to return Tensor ( #15904 )
...
matches jax
2026-04-24 08:27:17 -04:00
qazal
5eb641395a
viz/cli: select kernel events in -s DEV ( #15909 )
...
* simple test
* pass
2026-04-24 21:03:34 +09:00
nimlgen
c0f77c2e1c
hcq graph to linear ( #15888 )
...
* hcq
* f
* f
* linter
2026-04-24 12:42:49 +03:00
Christopher Milan
cbf4946ea6
usb: multiple gpus and better error messages ( #15900 )
2026-04-24 01:57:19 -04:00
chenyu
c24da99d56
avg_pool2d, max_pool2d to mixin ( #15903 )
...
* avg_pool2d, max_pool2d to mixin
* fix
* just dtype
* that
2026-04-23 23:36:17 -04:00
chenyu
08d9106c9f
scatter_reduce and sparse_categorical_crossentropy to mixin ( #15902 )
...
also use `.ne` to fix `# type: ignore[comparison-overlap]`
2026-04-23 21:06:36 -04:00
chenyu
8cc2c69e21
fix isclose mixin ( #15898 )
...
use `.eq` instead of `==`
2026-04-23 20:40:43 -04:00
chenyu
782bc6aece
broadcast in ElementwiseMixin.div [pr] ( #15897 )
2026-04-23 16:02:43 -04:00
qazal
7745e05a2f
sqtt: update wave end packet names ( #15896 )
...
* sqtt: update wave end packet names
* update wavestart and emu
2026-04-24 04:21:22 +09:00
chenyu
11c197955b
interpolate and cross_entropy to mixin ( #15895 )
2026-04-23 14:59:45 -04:00
chenyu
f0dbc68aa9
gather to mixin ( #15891 )
2026-04-23 14:00:57 -04:00
chenyu
87223f870e
logcumsumexp, argmax, argmin, sequential to mixin ( #15890 )
2026-04-23 12:10:42 -04:00
nimlgen
5cf4ad2fb6
fix resolve param ( #15889 )
2026-04-23 17:41:44 +03:00
George Hotz
0c3260d5d9
rename VECTORIZE to STACK ( #15880 )
2026-04-23 10:43:42 +08:00
chenyu
7c9bc29e44
Tensor method raise if arg is on different device ( #15879 )
...
instead of implicit `to`. this matches torch
2026-04-22 22:20:22 -04:00
chenyu
1fc4b3788c
cummax/cummin to mixin ( #15877 )
2026-04-22 21:25:39 -04:00
chenyu
684e95e1d4
UOp binary op broadcasts dtype ( #15875 )
...
* UOp binary op broadcasts dtype
matches Tensor
* fix
* fix?
2026-04-22 20:37:19 -04:00
Christopher Milan
b0dc95a390
AMX in arch, better docs ( #15871 )
2026-04-22 17:25:18 -04:00
nimlgen
e5891acab2
jit: precompile ( #15848 )
...
* x
* jit: precompile as sep step
* x
* s
* x
* x
* x
* ?
* ?
* x
* x
* viz
* f
* x
* u
* x
* x
2026-04-23 00:23:32 +03:00
chenyu
b9e2bc619e
simplify bool.cast() != const ( #15874 )
2026-04-22 17:08:09 -04:00
chenyu
3c8daa9a75
update test_where_removal ( #15872 )
...
don't use UOp.ufix for const_like, it will broadcast dtype soon
2026-04-22 14:56:37 -04:00
b1tg
af93a677ae
llm: glm 4.5 air ( #15771 )
...
* llm: glm 4.5 air
* clean
* clean
* remove gguf_size
2026-04-22 22:47:37 +08:00
qazal
719a7bdac5
viz: respect optional estimates in kernel info ( #15867 )
...
* simple failing test
* unpack kernel info
2026-04-22 14:24:48 +03:00
chenyu
3821e442eb
_one_hot_along_dim and one_hot to mixin ( #15861 )
2026-04-21 20:24:38 -04:00
chenyu
f911a63a6b
don't allow negative num_classes in one_hot ( #15859 )
...
no auto infer num_classes, matches jax
2026-04-21 19:39:29 -04:00
Christopher Milan
697e7aa819
MOCK+AMD and MOCK+NV interfaces ( #15858 )
...
MOCK+AMD is an alias for MOCKKFD+AMD, MOCKNVK+NV is renamed to MOCK+NV
2026-04-21 18:22:16 -04:00
chenyu
75ee51a446
triu tril _tri to mixin ( #15857 )
2026-04-21 17:10:55 -04:00
