@@ -119,16 +119,16 @@ def wrapped_set_pref_matmul(self, inputs, fp_accuracy):
119119
120120def 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