better deadlock report by pim simulator
This commit is contained in:
@@ -64,7 +64,11 @@ def main():
|
||||
ap.add_argument("--crossbar-size", type=int, default=64)
|
||||
ap.add_argument("--crossbar-count", type=int, default=8)
|
||||
ap.add_argument("--core-count", type=int, default=None,
|
||||
help="Core count to pass to Raptor. If omitted, Raptor uses its default.")
|
||||
help="Core count to pass to Raptor. Required for PIM validation.")
|
||||
ap.add_argument("--pim-merge-scheduler", choices=("peft", "dcp"), default="peft",
|
||||
help="Scheduler used by the Spatial merge-compute-nodes pass.")
|
||||
ap.add_argument("--command-timeout-seconds", type=float, default=60.0,
|
||||
help="Per-subprocess timeout in seconds for compiler, runner, and simulator commands.")
|
||||
ap.add_argument("--clean", action="store_true",
|
||||
help="Remove generated validation artifacts under each model workspace and exit.")
|
||||
ap.add_argument("--verbose", action="store_true",
|
||||
@@ -98,6 +102,8 @@ def main():
|
||||
missing_args.append("--raptor-path")
|
||||
if not a.onnx_include_dir:
|
||||
missing_args.append("--onnx-include-dir")
|
||||
if a.core_count is None:
|
||||
missing_args.append("--core-count")
|
||||
if missing_args:
|
||||
ap.error("the following arguments are required unless --clean is used: " + ", ".join(missing_args))
|
||||
|
||||
@@ -117,6 +123,8 @@ def main():
|
||||
result = validate_network(
|
||||
onnx_path, a.raptor_path, a.onnx_include_dir, simulator_dir,
|
||||
crossbar_size=a.crossbar_size, crossbar_count=a.crossbar_count, core_count=a.core_count,
|
||||
pim_merge_scheduler=a.pim_merge_scheduler,
|
||||
command_timeout_seconds=a.command_timeout_seconds,
|
||||
threshold=a.threshold,
|
||||
seed=a.seed,
|
||||
reporter=reporter,
|
||||
|
||||
Reference in New Issue
Block a user