add validation artifacts cleanup
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user