Skip to content

Update to TVM v0.6

Jonathan Mace requested to merge tvm-v0.6 into master

Updates our TVM code from v0.4 to v0.6. Tidies up some of the TVM dependency code too, but not all of it.

This is a "big" change, but it shouldn't break anything major. The main changes are in the compilation process from TVM models to Clockwork models; our Clockwork runtime only had a couple of changes, the tests all pass, and if anything breaks it'll be easy to track down the root cause.

@salzayat @rkarimi once this commit is merged, it will break your current environments. You'll need to pay attention to two things:

  1. Updating your TVM checkout
  2. Making sure you're using the correct model repository (if you're using one)

More specifically:

  • Concurrent with this change, I will merge changes to the tvm repository. You will need to pull those changes and rebuild TVM. This updates us to v0.6.
  • There's been a small tweak to our model format. Previously compiled models won't work without recompiling them. If you need this, talk to me; but I assume you aren't using any custom/personal models.
  • The out-of-the-box models in clockwork/resources have been updated and will work fine. If these are the only models you've been using, then you don't need to change anything.
  • Models in the clockwork-modelzoo-volta repository have been updated and will continue to work
  • Models in tvm-modelzoo no longer work, if you have been using those.
Edited by Jonathan Mace

Merge request reports