Skip to content

Commit

Permalink
add celu singularity function
Browse files Browse the repository at this point in the history
  • Loading branch information
beverlylytle committed Dec 5, 2024
1 parent b243842 commit 26fee91
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions thunder/tests/opinfos.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 26fee91

Please sign in to comment.