Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[tuner] Invoke iree-benchmark-module without calling bash scripts #99

Merged
merged 16 commits into from
Aug 22, 2024
Merged

[tuner] Invoke iree-benchmark-module without calling bash scripts #99

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
a17fe5d
Upgrade dispatch_benchmark
RattataKing Aug 20, 2024
0bcad45
Edit test_libtuner
RattataKing Aug 20, 2024
ba54305
Fix bugs in dispatch_benchmark()
RattataKing Aug 20, 2024
6ea5681
Upgrade model_benchmark()
RattataKing Aug 20, 2024
5abe57f
Maintain dry-run mode
RattataKing Aug 20, 2024
3c7a57d
Maintain pytest
RattataKing Aug 20, 2024
725a1fe
Small fix
RattataKing Aug 21, 2024
57caecb
Small fix
RattataKing Aug 21, 2024
58a745b
Upgrade compile_model
RattataKing Aug 21, 2024
145f768
Fix compile_unet commands
RattataKing Aug 21, 2024
b5b3aab
Fix lint
RattataKing Aug 21, 2024
8b707e9
Rearrange benchmarking completion detection. Format display strings
RattataKing Aug 21, 2024
c516eb9
Fix benchmark result parsing. Improve pytest
RattataKing Aug 21, 2024
01f36ee
Small fix
RattataKing Aug 21, 2024
5c44ea1
Fix punet commands
RattataKing Aug 22, 2024
ca1a44d
Clean up
RattataKing Aug 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions tuning/compile_candidate.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,6 @@ readonly DIR="$(dirname "$INPUT")"
readonly BASENAME="$(basename "$INPUT" .mlir)"
readonly OUT="${DIR}/compiled/${BASENAME}.vmfb"

mkdir -p "${DIR}/compiled" "${DIR}/failed" "${DIR}/specs"

timeout 4s ./punet.sh "$INPUT" -o "$OUT" --compile-from=executable-sources 2>/dev/null || (mv "$INPUT" "$DIR/failed" && exit 1)
tools/iree-dump-module "$OUT" | grep -q 'rocm-hsaco-fb' || (mv "$INPUT" "$DIR/failed" && rm -f "$OUT" && exit 1)
if [ -f "${DIR}/${BASENAME}_config.mlir" ]; then
Expand Down
Loading
Loading