chenyuandGitHub bdcdf1f1a1 jittable masked_select and nonzero (#16170) ...

* jittable masked_select and nonzero

make jittable with `size=`, matches jax

* COMPILE_ONLY

2026-05-12 16:39:36 -04:00

..

__init__.py

move tests to test/backend (#14691)

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

test_arange.py

cleaner and faster run_linear (#15987)

2026-04-30 20:15:22 +03:00

test_asm_gemm.py

update Tensor.allclose to return Tensor (#15904)

2026-04-24 08:27:17 -04:00

test_const_folding.py

remove schedule batch 3 (#15924)

2026-04-25 11:53:16 +03:00

test_custom_kernel.py

add all codegen stages to spec_tensor (#16163)

2026-05-12 10:35:38 -07:00

test_device.py

Device.count() (#15842)

2026-04-21 16:46:38 -04:00

test_dtype_alu.py

MOCKGPU interfaces (#15796)

2026-04-17 21:56:29 -04:00

test_dtype.py

deprecate <dev>=1 in favor of DEV=<dev> (#15467)

2026-03-26 03:48:03 -04:00

test_edgecases.py

fix empty input for masked_select and nonzero (#16168)

2026-05-12 15:36:51 -04:00

test_graph.py

validate_with_cpu as rewrite (#15938)

2026-04-26 19:58:53 +03:00

test_interop.py

MOCKGPU interfaces (#15796)

2026-04-17 21:56:29 -04:00

test_jit_cases.py

move tests to test/backend (#14691)

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

test_jit_footguns.py

jittable masked_select and nonzero (#16170)

2026-05-12 16:39:36 -04:00

test_jit.py

runtime/graph: avoid core_id runtimevar merge conflicts (#16026)

2026-05-03 19:16:02 +03:00

test_kernel_cache.py

move tests to test/backend (#14691)

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

test_linearizer_dumb.py

programinfo (#15942)

2026-04-27 23:12:03 +03:00

test_linearizer.py

split render to render.py (#16002)

2026-04-30 19:41:14 -07:00

test_multitensor.py

fix randperm and _multi_like drop requires_grad (#16150)

2026-05-11 23:23:34 -04:00

test_nn.py

Revert "[pr] match torch rmsnorm (#16122)" (#16144)

2026-05-11 17:53:42 -04:00

test_ops.py

jittable masked_select and nonzero (#16170)

2026-05-12 16:39:36 -04:00

test_opt_gemm.py

remove schedule() from tests batch 2 (#15923)

2026-04-25 10:44:41 +03:00

test_optim.py

move tests to test/backend (#14691)

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

test_pickle.py

remove schedule batch 3 (#15924)

2026-04-25 11:53:16 +03:00

test_profiler.py

Revert "cleanup: untrack wait Metal buffers (#15954)" (#16010)

2026-05-01 21:18:19 +09:00

test_quantize_onnx.py

programinfo (#15942)

2026-04-27 23:12:03 +03:00

test_randomness.py

no replacement multinomial (#15995)

2026-04-30 17:35:26 -04:00

test_rangeify.py

move tests to test/backend (#14691)

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

test_renderer_failures.py

remove CompiledRunner (#15970)

2026-04-29 22:45:48 +03:00

test_schedule.py

div to mixin (#16078)

2026-05-07 12:52:37 -04:00

test_setitem.py

Tensor.invalid -> Tesnor.invalids (#15849)

2026-04-21 11:19:51 -04:00

test_softmax_fusion.py

remove schedule() from tests batch 2 (#15923)

2026-04-25 10:44:41 +03: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

MOCKGPU interfaces (#15796)

2026-04-17 21:56:29 -04: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

explicit args to Tensor.rand (#16161)

2026-05-12 12:53:39 -04:00

test_to_numpy.py

move tests to test/backend (#14691)

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

test_transcendental.py

MOCKGPU interfaces (#15796)

2026-04-17 21:56:29 -04:00

test_uops.py

time_call (#16152)

2026-05-12 16:58:28 +03:00

test_zero_copy.py

move tests to test/backend (#14691)

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