From fdaec4b5067b92d9b16bac312ed40302d6a0637a Mon Sep 17 00:00:00 2001 From: Dave Liddell Date: Wed, 12 Jun 2024 00:01:43 -0700 Subject: [PATCH] Add plugin arg to vmfbRunner --- models/turbine_models/model_runner.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/models/turbine_models/model_runner.py b/models/turbine_models/model_runner.py index bdc81bcf8..d49fa8362 100644 --- a/models/turbine_models/model_runner.py +++ b/models/turbine_models/model_runner.py @@ -4,9 +4,13 @@ class vmfbRunner: - def __init__(self, device, vmfb_path, external_weight_path=None): + def __init__(self, device, vmfb_path, external_weight_path=None, extra_plugin=None): flags = [] - haldriver = ireert.get_driver(device) + clean_driver = False + if extra_plugin: + ireert.flags.parse_flags(f"--executable_plugin={extra_plugin}") + clean_driver = True + haldriver = ireert.get_driver(device, clean_driver) if "://" in device: try: device_idx = int(device.split("://")[-1])