huge refactor for high RewritePatterns usage and less ad-hoc cpp code
Validate Operations / validate-operations (push) Has been cancelled
Validate Operations / validate-operations (push) Has been cancelled
remove Spatial many ops in favor of tensor ops like in pim
This commit is contained in:
@@ -33,8 +33,6 @@ def _parse_pim_pass_timings(output_text):
|
||||
pass_timings[label] = pass_timings.get(label, 0.0) + duration
|
||||
break
|
||||
|
||||
if not pass_timings:
|
||||
raise RuntimeError("Raptor timing report did not contain any PIM pass timings.")
|
||||
return pass_timings
|
||||
|
||||
|
||||
@@ -43,7 +41,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, cwd=None, reporter=None):
|
||||
crossbar_size, crossbar_count, core_count=None, cwd=None, verbose=False, reporter=None):
|
||||
# Define the arguments, with the possibility to set crossbar size and count
|
||||
args = [
|
||||
network_path,
|
||||
@@ -51,13 +49,13 @@ def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
|
||||
output_base,
|
||||
"--maccel=PIM",
|
||||
"--EmitPimCodegen",
|
||||
# "--use-experimental-conv-impl=true",
|
||||
f"--crossbar-size={crossbar_size}",
|
||||
f"--crossbar-count={crossbar_count}",
|
||||
"--enable-timing",
|
||||
]
|
||||
if core_count is not None:
|
||||
args.append(f"--core-count={core_count}")
|
||||
if verbose:
|
||||
args.append("--enable-timing")
|
||||
|
||||
cmd = [str(raptor_onnx_path)] + [str(arg) for arg in args]
|
||||
if reporter is not None:
|
||||
|
||||
Reference in New Issue
Block a user