Teddy TennantandGitHub 1c183e6de9 fix elu/celu/selu gradient being nan on large inputs (#17852) ...

* fix elu/celu/selu gradient being nan on large inputs

* shrink shape for the elu/celu/selu overflow checks

* route selu and celu through elu

2026-08-30 15:12:52 -04: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

fix am_smi to respect dev (#17742)

2026-08-25 19:21:55 -07:00

test_assign.py

fix test_assign for no subbuffer backends (#17757)

2026-08-27 10:47:21 -04: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

test updates for weak const change (#17606)

2026-08-19 15:56:01 -04:00

test_custom_kernel.py

use KernelCountException when kernel count is being compared (#17584)

2026-08-18 16:06:03 -07:00

test_dtype_alu.py

skip INT_MIN % -1 on X86 and LLVM (#17637)

2026-08-20 17:26:26 -04:00

test_dtype.py

update X86 cast related stuff (#17816)

2026-08-28 14:30:34 -04:00

test_edgecases.py

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

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

test_encodings.py

put dtype to Ops.INS arg (#17817)

2026-08-28 15:05:17 -04: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

put dtype to Ops.INS arg (#17817)

2026-08-28 15:05:17 -04:00

test_jit.py

tiny hcq2 changes (#17797)

2026-08-28 11:29:00 +03:00

test_linearizer_dumb.py

UOp.param accepts single sint as shape (int,) (#17784)

2026-08-27 14:07:33 -07:00

test_linearizer.py

test updates for weak const change (#17606)

2026-08-19 15:56:01 -04:00

test_llama_kernels.py

hotfix: qkv bw kernel requires hipcc and cdna4 (#17645)

2026-08-21 12:38:45 +09:00

test_multitensor.py

hcq2 spec2 (#17798)

2026-08-28 12:51:55 +03:00

test_nn.py

fix Conv2d same padding with per axis dilation (#17805)

2026-08-28 12:44:43 -04:00

test_ops.py

fix elu/celu/selu gradient being nan on large inputs (#17852)

2026-08-30 15:12: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

fix Muon weight decay being a no-op (#17709)

2026-08-24 13:33:10 -07:00

test_pickle.py

use KernelCountException when kernel count is being compared (#17584)

2026-08-18 16:06:03 -07:00

test_profiler.py

remove copyin/copyout from Buffer (#17020)

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

test_quantize_onnx.py

delete dead DType.scalar [PR] (#17561)

2026-08-16 21:12:17 -04:00

test_randomness.py

DEFAULT_FLOAT/DEFAULT_INT ContextVar [pr] (#17265)

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

test_rangeify.py

delete PCONTIG [PR] (#17756)

2026-08-26 12:07:28 -07:00

test_renderer_failures.py

UOp.param accepts single sint as shape (int,) (#17784)

2026-08-27 14:07:33 -07:00

test_schedule.py

copy movement op tests construct copies (#17792)

2026-08-28 11:09:11 +09:00

test_setitem.py

delete unused rewrite rules [PR] (#17748)

2026-08-26 10:50:42 -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

cannot decomp long Variable (#17842)

2026-08-29 19:18:11 -04:00

test_tensor.py

add max_shape/max_numel to mixins + pad_to (#17553)

2026-08-15 20:05:10 -07: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

dtype is not a UOp field anymore (#17821)

2026-08-28 16:17:29 -04:00

test_wait_loop.py

UOp.param accepts single sint as shape (int,) (#17784)

2026-08-27 14:07:33 -07:00

test_zero_copy.py

move tests to test/backend (#14691)

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