Skip to content

Commit

Permalink
Edit iree binaries path
Browse files Browse the repository at this point in the history
  • Loading branch information
RattataKing committed Aug 28, 2024
1 parent 44e215f commit da9cc19
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion tuner/examples/punet/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,4 +32,4 @@ python punet_autotune.py 1286.mlir --num-candidates=64 --num-model-candidates=10
### Basic Usage
```
python punet_autotune.py 1286.mlir
```
```
6 changes: 3 additions & 3 deletions tuner/examples/punet/punet_autotune.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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",
Expand All @@ -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",
Expand Down

0 comments on commit da9cc19

Please sign in to comment.