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.

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

[Fix][Tutorial][VTA] Update tune_relay_vta.py to support single board #7100

Merged
merged 1 commit into from
Dec 14, 2020

Conversation

insop
Copy link
Contributor

@insop insop commented Dec 13, 2020

Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers by @ them in the pull request thread.

@insop
Copy link
Contributor Author

insop commented Dec 13, 2020

With this change on top of the latest TVM (28647f2), here is an example log with a single board.

$ python vta/tutorials/autotvm/tune_relay_vta.py
Extract tasks...
...100%, 0.06 MB, 174 KB/s, 0 seconds passed
...100%, 0.73 MB, 1274 KB/s, 0 seconds passed
Extracted 10 conv2d tasks:
(1, 14, 14, 256, 512, 1, 1, 0, 0, 2, 2)
(1, 28, 28, 128, 256, 1, 1, 0, 0, 2, 2)
(1, 56, 56, 64, 128, 1, 1, 0, 0, 2, 2)
(1, 56, 56, 64, 64, 3, 3, 1, 1, 1, 1)
(1, 28, 28, 128, 128, 3, 3, 1, 1, 1, 1)
(1, 56, 56, 64, 128, 3, 3, 1, 1, 2, 2)
(1, 14, 14, 256, 256, 3, 3, 1, 1, 1, 1)
(1, 28, 28, 128, 256, 3, 3, 1, 1, 2, 2)
(1, 7, 7, 512, 512, 3, 3, 1, 1, 1, 1)
(1, 14, 14, 256, 512, 3, 3, 1, 1, 2, 2)
Tuning...
[Task  1/10]  Current/Best:   17.63/ 116.35 GFLOPS | Progress: (480/480) | 255.47 s Done.
[Task  2/10]  Current/Best:   19.43/ 137.07 GFLOPS | Progress: (576/576) | 291.50 s Done.
[Task  3/10]  Current/Best:    0.00/ 137.87 GFLOPS | Progress: (672/1000) | 365.49 s

@insop
Copy link
Contributor Author

insop commented Dec 13, 2020

@tmoreau89

@insop
Copy link
Contributor Author

insop commented Dec 13, 2020

I found the change that I picked was already in the PR by created a separate file.

@insop
Copy link
Contributor Author

insop commented Dec 13, 2020

@i24361 , @merrymercy

@insop insop changed the title Update tune_relay_vta.py to support single board [Fix][Tutorial][VTA] Update tune_relay_vta.py to support single board Dec 13, 2020
Copy link
Contributor

@tmoreau89 tmoreau89 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for the fixes @insop and bringing this script up to date!

@tmoreau89 tmoreau89 merged commit 6be4d0a into apache:main Dec 14, 2020
@tmoreau89
Copy link
Contributor

The PR has been merged.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants