rework actually broken dcp merge + compute re-batching (still to refine)
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import re
|
||||
import shlex
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
from colorama import Fore, Style
|
||||
@@ -37,8 +38,12 @@ def _parse_pim_pass_timings(output_text):
|
||||
return pass_timings
|
||||
|
||||
|
||||
def _format_command(cmd):
|
||||
return shlex.join(str(arg) for arg in cmd)
|
||||
|
||||
|
||||
def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
|
||||
crossbar_size, crossbar_count, cwd=None, reporter=None):
|
||||
crossbar_size, crossbar_count, core_count=None, cwd=None, reporter=None):
|
||||
# Define the arguments, with the possibility to set crossbar size and count
|
||||
args = [
|
||||
network_path,
|
||||
@@ -51,10 +56,18 @@ def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
|
||||
f"--crossbar-count={crossbar_count}",
|
||||
"--enable-timing",
|
||||
]
|
||||
if core_count is not None:
|
||||
args.append(f"--core-count={core_count}")
|
||||
|
||||
cmd = [str(raptor_onnx_path)] + [str(arg) for arg in args]
|
||||
if reporter is not None:
|
||||
reporter.log(f" Raptor command: {_format_command(cmd)}")
|
||||
else:
|
||||
print(f"Raptor command: {_format_command(cmd)}")
|
||||
|
||||
try:
|
||||
output_text = run_command_with_reporter(
|
||||
[str(raptor_onnx_path)] + [str(arg) for arg in args],
|
||||
cmd,
|
||||
cwd=cwd,
|
||||
reporter=reporter,
|
||||
capture_output=True,
|
||||
|
||||
Reference in New Issue
Block a user