Skip to content

Commit dd8243a

Browse files
committed
number of warmup iters is now configurable + clean up lit config
1 parent fbcc453 commit dd8243a

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
config.excludes = ["README.md", "mlir_utils.py", "payload.py", "runner.py", "schedule.py"]
1+
config.excludes = ["mlir_utils.py", "payload.py", "runner.py", "schedule.py"]

python/examples/xegpu_matmul/matmul.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,12 @@ def parse_cli():
293293
default=1000,
294294
help="Number of runs to average the execution time.",
295295
)
296+
parser.add_argument(
297+
"--nwarmup",
298+
type=int,
299+
default=20,
300+
help="Number of warm-up iterations before benchmarking.",
301+
)
296302
parser.add_argument(
297303
"--relu",
298304
action="store_true",
@@ -375,7 +381,7 @@ def parse_cli():
375381
times = benchmark(
376382
wload,
377383
nruns=args.nruns,
378-
nwarmup=15,
384+
nwarmup=args.nwarmup,
379385
check_correctness=args.check_result,
380386
schedule_parameters=params,
381387
verbose=1,

0 commit comments

Comments
 (0)