From 3eb4e16c10ffc9de677a03157511727c0e80ce49 Mon Sep 17 00:00:00 2001
From: Amily Wu <amilywu2@amd.com>
Date: Wed, 28 Aug 2024 13:57:10 -0500
Subject: [PATCH] Edit iree binaries path

---
 tuner/examples/punet/punet_autotune.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tuner/examples/punet/punet_autotune.py b/tuner/examples/punet/punet_autotune.py
index 8bbdb4d16..a5dc407ea 100644
--- a/tuner/examples/punet/punet_autotune.py
+++ b/tuner/examples/punet/punet_autotune.py
@@ -47,7 +47,7 @@ def get_dispatch_benchmark_command(
         command = [
             "timeout",
             "16s",
-            "./tools/iree-benchmark-module",
+            "./iree-benchmark-module",
             f"--device={libtuner.DEVICE_ID_PLACEHOLDER}",
             f"--module={compiled_vmfb_path.resolve()}",
             "--hip_use_streams=true",
@@ -70,7 +70,7 @@ def get_model_compile_command(
             "timeout",
             "300s",
             "./compile-punet-base.sh",
-            "./tools/iree-compile",
+            "./iree-compile",
             "gfx942",
             f"{mlir_spec_path.resolve()}",
             "./punet.mlir",
@@ -88,7 +88,7 @@ def get_model_benchmark_command(
         command = [
             "timeout",
             "180s",
-            "tools/iree-benchmark-module",
+            "iree-benchmark-module",
             f"--device={libtuner.DEVICE_ID_PLACEHOLDER}",
             "--hip_use_streams=true",
             "--hip_allow_inline_execution=true",