You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by "junrushao (via GitHub)" <gi...@apache.org> on 2023/05/06 03:08:32 UTC

[GitHub] [tvm] junrushao opened a new pull request, #14784: [MetaSchedule] Introduce `tune_tir_module`

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

   This PR introduces a new user-facing API `tune_tir_module` that makes it more convenient to feed in an IRModule consists of multiple TIRs and customize the search space for them. This is widely used in our MLC-LLM project where we wanted to customize the search space for some quantization-related operators.
   
   An example usecase:
   
   ```python
   
   from tvm import meta_schedule as ms
   
   ms.tir_integration.tune_tir_module(
       ...
       space="cuda",  <========== by default, the space is "cuda" rather than "cuda-tensorcore"
       special_space={
           "fused_decode1_fused_matmul2_add1_gelu": sch_fused_decode_gemv,
           "decode": sch_decode,
       },
   )
   ```


-- 
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 #14784: [MetaSchedule] Enhance `tune_tir` to tune IRModule of TIR Collections

Posted by "junrushao (via GitHub)" <gi...@apache.org>.

junrushao merged PR #14784:
URL: https://github.com/apache/tvm/pull/14784


-- 
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 #14784: [MetaSchedule] Introduce `tune_tir_module`

Posted by "tvm-bot (via GitHub)" <gi...@apache.org>.

tvm-bot commented on PR #14784:
URL: https://github.com/apache/tvm/pull/14784#issuecomment-1537026514

   <!---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.
   
   
   
   <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