refactors
This commit is contained in:
@@ -15,8 +15,8 @@ REPO_ROOT = VALIDATION_DIR.parent
|
||||
if str(VALIDATION_DIR) not in sys.path:
|
||||
sys.path.insert(0, str(VALIDATION_DIR))
|
||||
|
||||
from onnx_utils import _ONNX_TO_NP, onnx_io, write_inputs_to_memory_bin
|
||||
from validate_one import (
|
||||
from raptor_validation.onnx_utils import _ONNX_TO_NP, onnx_io, write_inputs_to_memory_bin
|
||||
from raptor_validation.validate_one import (
|
||||
MODE_COMPILE_ONLY,
|
||||
build_dump_ranges,
|
||||
parse_pim_simulator_outputs,
|
||||
@@ -59,6 +59,15 @@ def ensure_local_artifacts(args, model_path: Path):
|
||||
crossbar_size=args.crossbar_size,
|
||||
crossbar_count=args.crossbar_count,
|
||||
core_count=args.core_count,
|
||||
raptor_extra_args=[],
|
||||
pimsim_nn_build_dir=None,
|
||||
pimsim_config_path=None,
|
||||
threshold=1e-3,
|
||||
rtol=1e-5,
|
||||
seed=0,
|
||||
reporter=None,
|
||||
model_index=1,
|
||||
model_total=1,
|
||||
command_timeout_seconds=args.command_timeout_seconds,
|
||||
mode=MODE_COMPILE_ONLY,
|
||||
verbose=args.verbose,
|
||||
|
||||
Reference in New Issue
Block a user