add small input shapes to some ops (#30617)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30617
as title
Test Plan: buck run //caffe2/benchmarks/operator_benchmark:benchmark_all_test -- --iterations 1 --operator add,as_strided,cat,chunk,fill,linear,matmul,split
Reviewed By: hl475
Differential Revision: D18764248
fbshipit-source-id: 510cf83542822acfa1b7b5e475b0cc7432f7ac19
diff --git a/benchmarks/operator_benchmark/pt/add_test.py b/benchmarks/operator_benchmark/pt/add_test.py
index 7499e08..945571c 100644
--- a/benchmarks/operator_benchmark/pt/add_test.py
+++ b/benchmarks/operator_benchmark/pt/add_test.py
@@ -21,6 +21,7 @@
add_short_configs = op_bench.config_list(
attr_names=["M", "N", "K"],
attrs=[
+ [1, 1, 1],
[64, 64, 64],
[64, 64, 128],
],
diff --git a/benchmarks/operator_benchmark/pt/as_strided_test.py b/benchmarks/operator_benchmark/pt/as_strided_test.py
index 53716b4..f1a522b 100644
--- a/benchmarks/operator_benchmark/pt/as_strided_test.py
+++ b/benchmarks/operator_benchmark/pt/as_strided_test.py
@@ -14,6 +14,7 @@
as_strided_configs_short = op_bench.config_list(
attr_names=["M", "N", "size", "stride", "storage_offset"],
attrs=[
+ [8, 8, (2, 2), (1, 1), 0],
[256, 256, (32, 32), (1, 1), 0],
[512, 512, (64, 64), (2, 2), 1],
],
diff --git a/benchmarks/operator_benchmark/pt/cat_test.py b/benchmarks/operator_benchmark/pt/cat_test.py
index 18ca6ce..3ba1426 100644
--- a/benchmarks/operator_benchmark/pt/cat_test.py
+++ b/benchmarks/operator_benchmark/pt/cat_test.py
@@ -14,6 +14,7 @@
cat_configs_short = op_bench.config_list(
attr_names=['M', 'N', 'K', 'dim'],
attrs=[
+ [1, 1, 1, 0],
[256, 512, 1, 0],
[512, 512, 2, 1],
],
diff --git a/benchmarks/operator_benchmark/pt/chunk_test.py b/benchmarks/operator_benchmark/pt/chunk_test.py
index 581301d..546ef58 100644
--- a/benchmarks/operator_benchmark/pt/chunk_test.py
+++ b/benchmarks/operator_benchmark/pt/chunk_test.py
@@ -14,6 +14,7 @@
chunk_short_configs = op_bench.config_list(
attr_names=["M", "N", "chunks"],
attrs=[
+ [8, 8, 2],
[256, 512, 2],
[512, 512, 2],
],
diff --git a/benchmarks/operator_benchmark/pt/fill_test.py b/benchmarks/operator_benchmark/pt/fill_test.py
index a3b15a4..5a162db 100644
--- a/benchmarks/operator_benchmark/pt/fill_test.py
+++ b/benchmarks/operator_benchmark/pt/fill_test.py
@@ -6,6 +6,7 @@
fill_short_configs = op_bench.config_list(
attr_names=["N"],
attrs=[
+ [1],
[1024],
[2048],
],
@@ -34,7 +35,7 @@
return self.input_one.fill_(10)
-op_bench.generate_pt_test(fill_short_configs + fill_long_configs,
+op_bench.generate_pt_test(fill_short_configs + fill_long_configs,
Fill_Benchmark)
diff --git a/benchmarks/operator_benchmark/pt/linear_test.py b/benchmarks/operator_benchmark/pt/linear_test.py
index 99542bb..f935ff7 100644
--- a/benchmarks/operator_benchmark/pt/linear_test.py
+++ b/benchmarks/operator_benchmark/pt/linear_test.py
@@ -14,6 +14,7 @@
linear_configs_short = op_bench.config_list(
attr_names=["N", "IN", "OUT"],
attrs=[
+ [1, 1, 1],
[4, 256, 128],
[16, 512, 256],
],
diff --git a/benchmarks/operator_benchmark/pt/matmul_test.py b/benchmarks/operator_benchmark/pt/matmul_test.py
index 30b3923..cdc5b24 100644
--- a/benchmarks/operator_benchmark/pt/matmul_test.py
+++ b/benchmarks/operator_benchmark/pt/matmul_test.py
@@ -12,6 +12,7 @@
mm_short_configs = op_bench.config_list(
attr_names=["M", "N", "K", "trans_a", "trans_b"],
attrs=[
+ [1, 1, 1, True, False],
[128, 128, 128, True, False],
[256, 256, 256, False, True],
],
diff --git a/benchmarks/operator_benchmark/pt/split_test.py b/benchmarks/operator_benchmark/pt/split_test.py
index 2902ee8..34c79a6 100644
--- a/benchmarks/operator_benchmark/pt/split_test.py
+++ b/benchmarks/operator_benchmark/pt/split_test.py
@@ -14,6 +14,7 @@
split_configs_short = op_bench.config_list(
attr_names=["M", "N", "parts"],
attrs=[
+ [8, 8, 2],
[256, 512, 2],
[512, 512, 2],
],
