File tree Expand file tree Collapse file tree 3 files changed +15
-10
lines changed Expand file tree Collapse file tree 3 files changed +15
-10
lines changed Original file line number Diff line number Diff line change @@ -31,16 +31,16 @@ cp ./dump-mmt/module_main_0_dispatch_0_rocm_hsaco_fb_benchmark.mlir test-benchma
31
31
### Recommended Trial Run
32
32
For an initial trial to test the tuning loop, use:
33
33
``` shell
34
- python -m tuner.examples.punet.punet_autotune test-benchmark.mlir --num-candidates=10
34
+ python -m tuner.examples.punet test-benchmark.mlir --num-candidates=10
35
35
```
36
36
37
37
### Dry Run Test
38
38
To perform a dry run (no GPU required), use:
39
39
``` shell
40
- python -m tuner.examples.punet.punet_autotune test-benchmark.mlir --num-candidates=64 --num-model-candidates=10 --dry-run
40
+ python -m tuner.examples.punet test-benchmark.mlir --num-candidates=64 --num-model-candidates=10 --dry-run
41
41
```
42
42
43
43
### Basic Usage
44
44
``` shell
45
- python -m tuner.examples.punet.punet_autotune test-benchmark.mlir
45
+ python -m tuner.examples.punet test-benchmark.mlir
46
46
```
Original file line number Diff line number Diff line change
1
+ # Copyright 2024 Advanced Micro Devices, Inc.
2
+ #
3
+ # Licensed under the Apache License v2.0 with LLVM Exceptions.
4
+ # See https://llvm.org/LICENSE.txt for license information.
5
+ # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
+
7
+ from . import punet_autotune
8
+
9
+ punet_autotune .main ()
Original file line number Diff line number Diff line change 7
7
"""
8
8
Sample Usage:
9
9
10
- python -m tuner.examples.punet.punet_autotune benchmark.mlir --lhs-dims=bmk --rhs-dims=bkn --tile-dims=*mnk --devices=hip://0,hip://1 --num-candidates=64
10
+ python -m tuner.examples.punet benchmark.mlir --lhs-dims=bmk --rhs-dims=bkn --tile-dims=*mnk --devices=hip://0,hip://1 --num-candidates=64
11
11
12
12
13
13
Recommended Trial Run:
14
14
15
- python -m tuner.examples.punet.punet_autotune benchmark.mlir --num-candidates=1
15
+ python -m tuner.examples.punet benchmark.mlir --num-candidates=1
16
16
17
17
18
18
Dry Run Test (no gpu requried):
19
19
20
- python -m tuner.examples.punet.punet_autotune benchmark.mlir --num-candidates=64 --num-model-candidates=10 --dry-run
20
+ python -m tuner.examples.punet benchmark.mlir --num-candidates=64 --num-model-candidates=10 --dry-run
21
21
22
22
"""
23
23
@@ -185,7 +185,3 @@ def main():
185
185
libtuner .logging .debug (candidate )
186
186
if args .verbose :
187
187
print (candidate )
188
-
189
-
190
- if __name__ == "__main__" :
191
- main ()
You can’t perform that action at this time.
0 commit comments