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

* llama: custom quantize_mxfp4+transpose kernel (codex)

* rename to cpp

* inline

* cleanup

* lds load_bf16x4

* more tests, add Estimates

2026-08-06 16:13:28 +09: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

revert wrong custom kernel fix (#17427)

2026-08-05 18:31:35 -07:00

test_dtype_alu.py

more shift tests (#17083)

2026-07-19 16:05:13 -04:00

test_dtype.py

skip test_float_to_fp8e4m3_extreme_values (#17228)

2026-07-26 21:01:51 -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

add new schedule tests + format better (#17402)

2026-08-04 18:46:38 -07: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 pow on extreme inputs (#17397)

2026-08-04 19:30:43 -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