Skip to content

Commit

Permalink
fix: add safety factors for test torch leaky relu and setup alphas co…
Browse files Browse the repository at this point in the history
…rrectly(#28392)
  • Loading branch information
aibenStunner authored Mar 1, 2024
1 parent f02ce81 commit 8fb4c26
Showing 1 changed file with 11 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -515,8 +515,18 @@ def test_torch_hardtanh_(
fn_tree="torch.nn.functional.leaky_relu",
dtype_and_x=helpers.dtype_and_values(
available_dtypes=helpers.get_dtypes("float"),
min_num_dims=1,
large_abs_safety_factor=25,
small_abs_safety_factor=25,
safety_factor_scale="log",
),
alpha=helpers.floats(
min_value=0,
max_value=1,
large_abs_safety_factor=25,
small_abs_safety_factor=25,
safety_factor_scale="log",
),
alpha=st.floats(min_value=0.0, max_value=1.0, exclude_min=True),
test_inplace=st.booleans(),
test_with_out=st.just(False),
)
Expand Down

0 comments on commit 8fb4c26

Please sign in to comment.