Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
beverlylytle committed Nov 8, 2024
1 parent cacb115 commit 9f6a196
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions thunder/tests/opinfos.py
Original file line number Diff line number Diff line change
Expand Up @@ -1633,7 +1633,7 @@ def _abs_torch(x: torch.Tensor | Number):
elementwise_unary_ops.append(reciprocal_opinfo)


def elementise_unary_with_alpha_generator(op, device, dtype, requires_grad):
def elementwise_unary_with_alpha_generator(op, device, dtype, requires_grad):
alphas = (None, -1.0, 0.5)
samples = elementwise_unary_generator(op, device, dtype, requires_grad)
for alpha, sample in itertools.product(alphas, samples):
Expand All @@ -1646,7 +1646,7 @@ def elementise_unary_with_alpha_generator(op, device, dtype, requires_grad):
celu_opinfo = OpInfo(
ltorch.celu,
dtypes=(datatypes.floating,),
sample_input_generator=elementise_unary_with_alpha_generator,
sample_input_generator=elementwise_unary_with_alpha_generator,
torch_reference=_elementwise_unary_torch(torch.celu),
test_directives=(),
)
Expand All @@ -1656,7 +1656,7 @@ def elementise_unary_with_alpha_generator(op, device, dtype, requires_grad):
elu_opinfo = OpInfo(
ltorch.elu,
dtypes=(datatypes.floating,),
sample_input_generator=elementise_unary_with_alpha_generator,
sample_input_generator=elementwise_unary_with_alpha_generator,
torch_reference=torch.nn.functional.elu,
# fdm.jvp, which is used in test_vjp_correctness, behaves badly on (-1e-6, 1e-6) for this function
singularity_fn=lambda x: x,
Expand Down

0 comments on commit 9f6a196

Please sign in to comment.