wozeparrot 674c760974 embedded bwd vocab shard (#15001) ...

* fix: remove more multi from call

* feat: embedding bwd vocab sharding

* clean: unused import

* clean: don't actually need this pattern

2026-03-16 19:37:16 -07:00

..

__init__.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_arange.py

embedded bwd vocab shard (#15001)

2026-03-16 19:37:16 -07:00

test_asm_gemm.py

sqtt: add CDNA ops enum, show in viz (#15140)

2026-03-17 09:38:42 +09:00

test_const_folding.py

fix copying padded const (#15116)

2026-03-04 10:39:45 +08:00

test_custom_kernel.py

all consts have shapes (#14959)

2026-02-23 10:26:50 +08:00

test_dtype_alu.py

remove staticmethod dtypes.max/min (#15227)

2026-03-11 23:11:24 -04:00

test_dtype.py

remove staticmethod dtypes.max/min (#15227)

2026-03-11 23:11:24 -04:00

test_edgecases.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_graph.py

more graph tests (#15130)

2026-03-04 19:01:12 +03:00

test_image_dtype.py

preallocate all realized buffers (#14823)

2026-02-20 20:05:54 +08:00

test_interop.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_jit_cases.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_jit_footguns.py

jit: fix raw in same kernel (#14699)

2026-02-12 15:33:32 +03:00

test_jit.py

update test_jit_init_empty (#14938)

2026-02-21 09:01:50 -05:00

test_kernel_cache.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_linearizer_dumb.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_linearizer.py

move test_grouped_dims to test/null (#14893)

2026-02-19 14:50:53 -05:00

test_multitensor.py

update test_multiple_to_single_device (#15056)

2026-02-27 21:44:33 -05:00

test_nn.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_ops.py

remove staticmethod dtypes.max/min (#15227)

2026-03-11 23:11:24 -04:00

test_opt_gemm.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_optim.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_pickle.py

update test to reset and test kernel_count directly (#14832)

2026-02-17 11:48:46 -05:00

test_profiler.py

test_profiler cleanup, non flaky cpu_profile test (#14932)

2026-02-21 16:58:10 +09:00

test_quantize_onnx.py

remove staticmethod dtypes.max/min (#15227)

2026-03-11 23:11:24 -04:00

test_randomness.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_rangeify.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_renderer_failures.py

remove staticmethod dtypes.max/min (#15227)

2026-03-11 23:11:24 -04:00

test_schedule.py

start using Invalid in image_conv2d (#15208)

2026-03-10 07:11:06 -04:00

test_setitem.py

update assign tests to also test the expected behavior (#15132)

2026-03-04 11:34:43 -05:00

test_softmax_fusion.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_stunning.py

raise RuntimeError in schedule for conflicted var_val [pr] (#15031)

2026-02-26 15:16:01 -05:00

test_subbuffer.py

buffer view is like buffer (#15082)

2026-03-03 09:52:33 +08:00

test_symbolic_jit.py

allocate generates a call (#14958)

2026-02-23 15:59:20 +08:00

test_symbolic_ops.py

Support triu variable on diagonal + SDPA symbolic (#15081)

2026-03-02 12:19:48 +08:00

test_tensor_variable.py

Support triu variable on diagonal + SDPA symbolic (#15081)

2026-03-02 12:19:48 +08:00

test_tensor.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_to_numpy.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00

test_transcendental.py

enable a few WEBGPU isnan tests that work now (#14967)

2026-02-23 11:06:08 -05:00

test_uops.py

enable a few WEBGPU isnan tests that work now (#14967)

2026-02-23 11:06:08 -05:00

test_zero_copy.py

move tests to test/backend (#14691)

2026-02-12 11:09:44 +08:00