From 27c56e9332e82da4b8339b205ec796cfd1a03e41 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Wed, 18 Feb 2026 10:06:38 +0000 Subject: [PATCH] activation: enable CI tests --- activation/flake.lock | 6 +++--- activation/tests/kernels/test_activation.py | 2 ++ 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/activation/flake.lock b/activation/flake.lock index d3a54b28..966d8758 100644 --- a/activation/flake.lock +++ b/activation/flake.lock @@ -41,11 +41,11 @@ "rust-overlay": "rust-overlay" }, "locked": { - "lastModified": 1769780418, - "narHash": "sha256-h8FhylXMFqxnH750D9Bq7QsqinBTVV7BdaJbHyWxwYY=", + "lastModified": 1771401433, + "narHash": "sha256-E6qyXe9xd8KuOTP1x/v/c+5QzhlzkGoOtHMnhwWs9/c=", "owner": "huggingface", "repo": "kernels", - "rev": "00075f6d86554eab25624755a030fe23a9a7bb06", + "rev": "80058c95e11862ce88aa9b8e52c575dbed6cc26f", "type": "github" }, "original": { diff --git a/activation/tests/kernels/test_activation.py b/activation/tests/kernels/test_activation.py index 684783db..f37e8a63 100644 --- a/activation/tests/kernels/test_activation.py +++ b/activation/tests/kernels/test_activation.py @@ -64,6 +64,7 @@ def gelu_tanh(x: torch.Tensor) -> torch.Tensor: def silu(x: torch.Tensor) -> torch.Tensor: return F.silu(x) +@pytest.mark.kernels_ci @pytest.mark.parametrize( "activation_name", ["silu_and_mul", "mul_and_silu", "gelu", "gelu_tanh", "fatrelu"] ) @@ -132,6 +133,7 @@ def test_act_and_mul( opcheck(op, (out, x)) +@pytest.mark.kernels_ci @pytest.mark.parametrize( "activation_fns", [