add validation artifacts cleanup

This commit is contained in:
NiccoloN
2026-03-20 13:15:08 +01:00
parent db3f52a647
commit 916a09414c
3 changed files with 59 additions and 17 deletions

View File

@@ -4,10 +4,13 @@ from colorama import Fore, Style
from subprocess_utils import run_command_with_reporter
def compile_with_raptor(network_path, raptor_onnx_path: Path, crossbar_size, crossbar_count, reporter=None):
def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
crossbar_size, crossbar_count, cwd=None, reporter=None):
# Define the arguments, with the possibility to set crossbar size and count
args = [
network_path,
"-o",
output_base,
"--maccel=PIM",
"--EmitPimCodegen",
# "--use-experimental-conv-impl=true",
@@ -18,6 +21,7 @@ def compile_with_raptor(network_path, raptor_onnx_path: Path, crossbar_size, cro
try:
run_command_with_reporter(
[str(raptor_onnx_path)] + [str(arg) for arg in args],
cwd=cwd,
reporter=reporter,
)
if reporter is None: