From 26fee91d875ebe54046c0229dccdf3fd8e357e7a Mon Sep 17 00:00:00 2001 From: beverlylytle Date: Thu, 5 Dec 2024 15:39:02 +0200 Subject: [PATCH] add celu singularity function --- thunder/tests/opinfos.py | 1 + 1 file changed, 1 insertion(+) diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py index d41a9ae433..81ddffe369 100644 --- a/thunder/tests/opinfos.py +++ b/thunder/tests/opinfos.py @@ -1649,6 +1649,7 @@ def gen(op, device, dtype, requires_grad): dtypes=(datatypes.floating,), sample_input_generator=get_elementwise_unary_with_alpha_generator(), torch_reference=_elementwise_unary_torch(torch.celu), + singularity_fn=lambda x: x, test_directives=(), ) elementwise_unary_ops.append(celu_opinfo)