Skip to content

Commit 85fab6c

Browse files
jainapurvapytorchmergebot
authored andcommitted
Fix duplicate benchmarking entries for addmm (pytorch#166652)
There have been duplicate entries for addmm in dashboard. This PR fixes the duplicate entries issues Pull Request resolved: pytorch#166652 Approved by: https://github.com/yangw-dev
1 parent c08ce30 commit 85fab6c

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

benchmarks/operator_benchmark/pt/addmm_test.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -53,10 +53,8 @@ def forward(self, input_one, mat1, mat2):
5353
return torch.addmm(input_one, mat1, mat2)
5454

5555

56-
op_bench.generate_pt_test(addmm_long_configs + addmm_long_configs, AddmmBenchmark)
57-
op_bench.generate_pt_gradient_test(
58-
addmm_long_configs + addmm_long_configs, AddmmBenchmark
59-
)
56+
op_bench.generate_pt_test(addmm_short_configs + addmm_long_configs, AddmmBenchmark)
57+
op_bench.generate_pt_gradient_test(addmm_long_configs, AddmmBenchmark)
6058

6159
"""Mircobenchmark for addbmm operator."""
6260

@@ -107,9 +105,7 @@ def forward(self, input_one, batch1, batch2):
107105
)
108106

109107
op_bench.generate_pt_test(addbmm_long_configs + addbmm_short_configs, AddbmmBenchmark)
110-
op_bench.generate_pt_gradient_test(
111-
addbmm_long_configs + addbmm_short_configs, AddbmmBenchmark
112-
)
108+
op_bench.generate_pt_gradient_test(addbmm_long_configs, AddbmmBenchmark)
113109

114110
if __name__ == "__main__":
115111
op_bench.benchmark_runner.main()

0 commit comments

Comments
 (0)