From dd4269e09f2fda5454a23d7e69e66d7f692c2c3d Mon Sep 17 00:00:00 2001 From: zjgarvey Date: Thu, 24 Oct 2024 19:12:22 -0500 Subject: [PATCH] Revert to using target-backends flag. Don't modify user inputs. --- alt_e2eshark/e2e_testing/backends.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/alt_e2eshark/e2e_testing/backends.py b/alt_e2eshark/e2e_testing/backends.py index 0efbbd7e..d2a4f4d6 100644 --- a/alt_e2eshark/e2e_testing/backends.py +++ b/alt_e2eshark/e2e_testing/backends.py @@ -34,9 +34,6 @@ class SimpleIREEBackend(BackendBase): '''This backend uses iree to compile and run MLIR modules for a specified hal_target_backend''' def __init__(self, *, device="local-task", hal_target_backend="llvm-cpu", extra_args : List[str] = None): self.device = device - if hal_target_backend == "hip": - print("IREE compiler python bindings do not currently support the change to using iree-hal-target-device=hip. Defaulting to deprecated iree-hal-target-backends=rocm") - hal_target_backend = "rocm" self.hal_target_backend = hal_target_backend self.extra_args = [] if extra_args: @@ -81,9 +78,6 @@ class CLIREEBackend(BackendBase): '''This backend calls iree through the command line to compile and run MLIR modules''' def __init__(self, *, device="local-task", hal_target_backend="llvm-cpu", extra_args : List[str] = None): self.device = device - if hal_target_backend == "rocm": - print("Using 'iree-hal-target-device=hip', since 'iree-hal-target-backends' is deprecated") - hal_target_backend = "hip" self.hal_target_device = hal_target_backend self.extra_args = [] if extra_args: @@ -95,7 +89,7 @@ def __init__(self, *, device="local-task", hal_target_backend="llvm-cpu", extra_ def compile(self, module_path: str, *, save_to : str = None) -> str: vmfb_path = os.path.join(save_to, "compiled_model.vmfb") - arg_string = f"--iree-hal-target-device={self.hal_target_device} " + arg_string = f"--iree-hal-target-backends={self.hal_target_device} " for arg in self.extra_args: arg_string += arg arg_string += " "