qazalandgeohot 32b9149040 llama: custom silu kernels (#17462) ...

* start by copying the C

* uop kernel

* cleanup tests

* estimates is part of SPEC

2026-08-10 14:06:15 +00:00

..

__init__.py

move tests to test/backend (#14691)

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

test_arange.py

more KernelCountException (#17407)

2026-08-04 22:55:38 -07:00

test_asm_gemm.py

llama: custom quantize_mxfp4+transpose kernel (codex) (#17434)

2026-08-06 16:13:28 +09:00

test_call.py

const(value, dtype) -> const(value).cast(dtype) in tests (#17335)

2026-07-31 13:24:31 -04:00

test_const_folding.py

const(dtype, b) -> const(b, dtype) [PR] (#17328)

2026-07-31 09:46:37 -04:00

test_custom_kernel.py

remove contiguous from custom_kernel (#17149)

2026-08-07 11:20:59 +09:00

test_dtype_alu.py

truncate float in DType.const [pr] (#17439)

2026-08-06 20:19:08 -04:00

test_dtype.py

fix f2f from fp8e5m2fnuz to half (#17442)

2026-08-06 21:19:46 -04:00

test_edgecases.py

Tensor.train -> TRAINING [PR] (#16705)

2026-06-22 15:13:22 -04:00

test_encodings.py

full removal of dtype.vec (#16996)

2026-07-12 09:25:50 -07:00

test_graph.py

remove copyin/copyout from Buffer (#17020)

2026-07-14 19:47:22 +03:00

test_interop.py

remove getenv(CI) (#16365)

2026-05-25 20:23:33 -04:00

test_isel.py

more const arg -> val (#17350)

2026-08-01 02:18:10 -04:00

test_jit.py

add new schedule tests + format better (#17402)

2026-08-04 18:46:38 -07:00

test_linearizer_dumb.py

const(value, dtype) -> const(value).cast(dtype) in tests (#17335)

2026-07-31 13:24:31 -04:00

test_linearizer.py

use check_schedule in tests where possible (#17400)

2026-08-04 18:17:29 -07:00

test_llama_kernels.py

llama: custom silu kernels (#17462)

2026-08-10 14:06:15 +00:00

test_multitensor.py

add new schedule tests + format better (#17402)

2026-08-04 18:46:38 -07:00

test_nn.py

use check_schedule in tests where possible (#17400)

2026-08-04 18:17:29 -07:00

test_ops.py

fix prod gradients at zero (#17404)

2026-08-07 09:56:52 -04:00

test_opt_gemm.py

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

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

test_optim.py

DEFAULT_FLOAT/DEFAULT_INT ContextVar [pr] (#17265)

2026-07-28 19:08:18 -04:00

test_pickle.py

const(value, dtype) -> const(value).cast(dtype) in tests (#17335)

2026-07-31 13:24:31 -04:00

test_profiler.py

remove copyin/copyout from Buffer (#17020)

2026-07-14 19:47:22 +03:00

test_quantize_onnx.py

remove Ops.DEVICE from Ops.PROGRAM [PR] (#16797)

2026-06-29 17:41:31 -04:00

test_randomness.py

DEFAULT_FLOAT/DEFAULT_INT ContextVar [pr] (#17265)

2026-07-28 19:08:18 -04:00

test_rangeify.py

data_src concept in run_rangeify [pr] (#17183)

2026-07-24 14:20:41 -04:00

test_renderer_failures.py

use check_schedule in tests where possible (#17400)

2026-08-04 18:17:29 -07:00

test_schedule.py

add new schedule tests + format better (#17402)

2026-08-04 18:46:38 -07:00

test_setitem.py

limit CONST(UNIQUE) to invalids only (#16432)

2026-05-29 16:02:06 -04:00

test_softmax_fusion.py

use check_schedule in tests where possible (#17400)

2026-08-04 18:17:29 -07:00

test_stunning.py

Tensor.train -> TRAINING [PR] (#16705)

2026-06-22 15:13:22 -04:00

test_subbuffer.py

remove copyin/copyout from Buffer (#17020)

2026-07-14 19:47:22 +03:00

test_symbolic_jit.py

allocate generates a call (#14958)

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

test_symbolic_ops.py

fix slice + non contig kernels (#17423)

2026-08-05 16:18:17 -07:00

test_tensor_variable.py

64-bit UOp.variable support, try 2 (#17256)

2026-07-29 14:39:36 -04:00

test_tensor.py

const(value, dtype) -> const(value).cast(dtype) in tests (#17335)

2026-07-31 13:24:31 -04:00

test_to_numpy.py

move tests to test/backend (#14691)

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

test_transcendental.py

use check_schedule in tests where possible (#17400)

2026-08-04 18:17:29 -07:00

test_uops.py

const(value, dtype) -> const(value).cast(dtype) in tests (#17335)

2026-07-31 13:24:31 -04:00

test_wait_loop.py

hotfix: revert test_scalar_alu_index, violates spec

2026-08-03 15:25:51 -07:00

test_zero_copy.py

move tests to test/backend (#14691)

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