Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
justinchuby committed Sep 24, 2024
1 parent 11aaacf commit 41909e3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/function_libs/torch_lib/extra_opinfo.py
Original file line number Diff line number Diff line change
Expand Up @@ -1855,12 +1855,12 @@ def shape(size, rank, with_batch_channel=True):
)
yield opinfo_core.SampleInput(
make_arg(shape(D, rank)),
shape(L, rank, False),
None,
[1.7, 1.5, 2.0], # scaler
)
yield opinfo_core.SampleInput(
make_arg(shape(D, rank)),
shape(L, rank, False),
None,
[0.6, 0.3, 0.5],
)

Expand Down

0 comments on commit 41909e3

Please sign in to comment.