better reports cleanups
This commit is contained in:
@@ -11,7 +11,6 @@ PIM_PASS_LABELS = (
|
||||
("SpatialToPimPass", "Spatial to PIM"),
|
||||
("PimBufferizationPass", "Bufferize PIM"),
|
||||
("HostConstantFoldingPass", "Fold Host Constants"),
|
||||
("PimMemoryCoalescingPass", "Coalesce Local Memory"),
|
||||
("PimLocalMemoryPlanningPass", "Plan Local Memory"),
|
||||
("VerificationPass", "Verify PIM"),
|
||||
("EmitPimCodePass", "Emit PIM Code"),
|
||||
@@ -43,7 +42,7 @@ def _format_command(cmd):
|
||||
|
||||
def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
|
||||
crossbar_size, crossbar_count, core_count=None,
|
||||
pim_memory_report="none", raptor_extra_args=None, cwd=None, verbose=False,
|
||||
raptor_extra_args=None, cwd=None, verbose=False,
|
||||
reporter=None, timeout_sec=None):
|
||||
# Define the arguments, with the possibility to set crossbar size and count
|
||||
args = [
|
||||
@@ -57,8 +56,6 @@ def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
|
||||
]
|
||||
if core_count is not None:
|
||||
args.append(f"--core-count={core_count}")
|
||||
if pim_memory_report != "none":
|
||||
args.append(f"--pim-memory-report={pim_memory_report}")
|
||||
if raptor_extra_args:
|
||||
args.extend(str(arg) for arg in raptor_extra_args)
|
||||
if verbose:
|
||||
|
||||
Reference in New Issue
Block a user