File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -466,7 +466,11 @@ def run_iree_compile_command(compile_pack: CompilePack) -> Optional[int]:
466
466
timeout_seconds = compile_pack .iree_compile_timeout ,
467
467
)
468
468
)
469
- if result .process_res is None or result .is_timeout :
469
+
470
+ # We need to check if the output vmfb exists as iree-compile returns a success
471
+ # status code when crash reproducers are dumped.
472
+ output_vmfb_exists = candidate_tracker .compiled_vmfb_path .is_file ()
473
+ if result .process_res is None or result .is_timeout or not output_vmfb_exists :
470
474
return None
471
475
return candidate_tracker .candidate_id
472
476
@@ -520,7 +524,7 @@ def run_iree_benchmark_module_command(benchmark_pack: BenchmarkPack):
520
524
** extra_flags ,
521
525
)
522
526
except ireert .benchmark .BenchmarkTimeoutError as e :
523
- logging .warning (
527
+ logging .info (
524
528
f"Benchmark of candidate { candidate_id } timed out after { timeout } seconds."
525
529
)
526
530
return BenchmarkResult (
@@ -557,7 +561,9 @@ def run_iree_benchmark_module_command(benchmark_pack: BenchmarkPack):
557
561
)
558
562
559
563
mean_benchmark_time = sum (times ) / float (len (times ))
560
- logging .debug (f"Benchmark time of candidate { candidate_id } : { mean_benchmark_time } " )
564
+ logging .debug (
565
+ f"Benchmark time of candidate { candidate_id } : { mean_benchmark_time :.2f} "
566
+ )
561
567
return BenchmarkResult (
562
568
candidate_id = candidate_id ,
563
569
time = mean_benchmark_time ,
You can’t perform that action at this time.
0 commit comments