change input shape to reduce variation (#27548)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/27548
as title
Test Plan: i_dont_want_it
Reviewed By: hl475
Differential Revision: D17811295
fbshipit-source-id: 3be957f6f3eaa464ebf4f5bd7c07d096ae4eae8c
diff --git a/benchmarks/operator_benchmark/pt/add_test.py b/benchmarks/operator_benchmark/pt/add_test.py
index e37ab10..3be7fb2 100644
--- a/benchmarks/operator_benchmark/pt/add_test.py
+++ b/benchmarks/operator_benchmark/pt/add_test.py
@@ -19,8 +19,8 @@
add_short_configs = op_bench.config_list(
attrs=[
- [32, 64, 64],
[64, 64, 64],
+ [64, 64, 128],
],
attr_names=["M", "N", "K"],
tags=["short"],