Skip to content

Commit 06d72c0

Browse files
authored
unify the opname of test for pad (#947)
1 parent c0d3f16 commit 06d72c0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

benchmark/test_special_perf.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def multinomial_input_fn(shape, dtype, device):
149149

150150
@pytest.mark.pad
151151
def test_perf_pad():
152-
def padding_input_fn(shape, dtype, device):
152+
def pad_input_fn(shape, dtype, device):
153153
input = torch.randn(shape, device=device, dtype=dtype)
154154
rank = input.ndim
155155
pad_params = [random.randint(0, 10) for _ in range(rank * 2)]
@@ -161,8 +161,8 @@ def padding_input_fn(shape, dtype, device):
161161
},
162162

163163
bench = GenericBenchmark(
164-
input_fn=padding_input_fn,
165-
op_name="padding",
164+
input_fn=pad_input_fn,
165+
op_name="pad",
166166
torch_op=torch.nn.functional.pad,
167167
dtypes=FLOAT_DTYPES,
168168
)

0 commit comments

Comments
 (0)