Skip to content

Commit fe542c3

Browse files
committed
New defaults for matmul benchmarks
1 parent d60eebf commit fe542c3

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

bench/ndarray/matmul_path_compare.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -119,16 +119,16 @@ def wrapped_set_pref_matmul(self, inputs, fp_accuracy):
119119

120120
def main() -> None:
121121
parser = argparse.ArgumentParser(description="Compare chunked and fast blosc2.matmul paths.")
122-
parser.add_argument("--shape-a", default="1000,1000", help="Comma-separated shape for A.")
123-
parser.add_argument("--shape-b", default="1000,1000", help="Comma-separated shape for B.")
124-
parser.add_argument("--dtype", default="float64", choices=["float32", "float64", "int32", "int64"])
125-
parser.add_argument("--chunks-a", default="500,500", help="Comma-separated chunk shape for A.")
126-
parser.add_argument("--chunks-b", default="500,500", help="Comma-separated chunk shape for B.")
122+
parser.add_argument("--shape-a", default="400,400", help="Comma-separated shape for A.")
123+
parser.add_argument("--shape-b", default="400,400", help="Comma-separated shape for B.")
124+
parser.add_argument("--dtype", default="float32", choices=["float32", "float64", "int32", "int64"])
125+
parser.add_argument("--chunks-a", default="200,200", help="Comma-separated chunk shape for A.")
126+
parser.add_argument("--chunks-b", default="200,200", help="Comma-separated chunk shape for B.")
127127
parser.add_argument("--blocks-a", default="100,100", help="Comma-separated block shape for A.")
128128
parser.add_argument("--blocks-b", default="100,100", help="Comma-separated block shape for B.")
129-
parser.add_argument("--chunks-out", default="500,500", help="Comma-separated chunk shape for output.")
129+
parser.add_argument("--chunks-out", default="200,200", help="Comma-separated chunk shape for output.")
130130
parser.add_argument("--blocks-out", default="100,100", help="Comma-separated block shape for output.")
131-
parser.add_argument("--repeats", type=int, default=5)
131+
parser.add_argument("--repeats", type=int, default=250)
132132
parser.add_argument("--modes", nargs="+", default=["chunked", "fast", "auto"], choices=["chunked", "fast", "auto"])
133133
parser.add_argument("--json", action="store_true", help="Emit full JSON instead of a compact text summary.")
134134
args = parser.parse_args()

0 commit comments

Comments
 (0)