You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2022/11/21 14:39:31 UTC

[GitHub] [tvm] yelite opened a new pull request, #13452: [MetaSchedule] TorchBench tuning script: add task extraction mode

yelite opened a new pull request, #13452:
URL: https://github.com/apache/tvm/pull/13452

   This PR adds the task extraction mode to the TorchBench tuning script. The task extraction mode will only import PyTorch model into TVM, extract tuning tasks from it and save them for further use. 
   
   This helps the use case where tasks should be extracted from a machine with GPU and then be tuned on a machine with powerful CPU with rpc workers.
   
   cc @junrushao @zxybazh 


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] junrushao merged pull request #13452: [MetaSchedule] TorchBench tuning script: add task extraction mode

Posted by GitBox <gi...@apache.org>.
junrushao merged PR #13452:
URL: https://github.com/apache/tvm/pull/13452


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] zxybazh commented on a diff in pull request #13452: [MetaSchedule] TorchBench tuning script: add task extraction mode

Posted by GitBox <gi...@apache.org>.
zxybazh commented on code in PR #13452:
URL: https://github.com/apache/tvm/pull/13452#discussion_r1029749134


##########
python/tvm/meta_schedule/testing/torchbench/run.py:
##########
@@ -124,28 +125,37 @@
 class RunMode(Enum):
     """
     The running mode of this script. Available values are:
-    - tune: Only tune the model and create the tuning database.
+    - extract: Only import the model and extract tuning tasks from it.
+    - tune: Only tune the tasks and create the tuning database.

Review Comment:
   Is it possible that we need some other combination of stages, say `extract+tune` or `tune+evaluate`?



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] tvm-bot commented on pull request #13452: [MetaSchedule] TorchBench tuning script: add task extraction mode

Posted by GitBox <gi...@apache.org>.
tvm-bot commented on PR #13452:
URL: https://github.com/apache/tvm/pull/13452#issuecomment-1322159827

   <!---bot-comment-->
   
   Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from [Reviewers](https://github.com/apache/incubator-tvm/blob/master/CONTRIBUTORS.md#reviewers) by @-ing them in a comment.
   
   <!--bot-comment-ccs-start-->
    * cc @Hzfengsy, @elvin-n, @junrushao <sub>See [#10317](https://github.com/apache/tvm/issues/10317) for details</sub><!--bot-comment-ccs-end-->
   
   <sub>Generated by [tvm-bot](https://github.com/apache/tvm/blob/main/ci/README.md#github-actions)</sub>


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] junrushao commented on pull request #13452: [MetaSchedule] TorchBench tuning script: add task extraction mode

Posted by GitBox <gi...@apache.org>.
junrushao commented on PR #13452:
URL: https://github.com/apache/tvm/pull/13452#issuecomment-1322529120

   CC: @zxybazh 


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org