[GitHub] [tvm] MasterJH5574 opened a new pull request, #13663: [TIR] Create Layout with specified axis dtype

[GitBox <gi...@apache.org>]

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

   This PR brings the support for specifying axis dtype when creating `tir::Layout`.
   
   Prior to this PR, the construction of `tir::Layout` uses int32 for every axis. With this specification, the output of `BijectiveLayout::ForwardShape/BackwardShape(...)` will always be int32 shape, no matter if the input shape is in int32 or int64.
   
   As we will be gradually moving to int64 default dtype for tensor shapes, there are increasingly more cases where we need to handle int64 shapes. Thus, `tir::Layout` need some updates for int64 shapes. Since the creation of an `tir::Layout` instance is agnostic to the dtype of the shapes that the instance will handle later, one way of the such updates is to add a dtype parameter to the constructor of `tir::Layout`, to specify the dtype explicitly. This PR enhances `tir::Layout` in this way, with appropriate unit tests for dtype check.
   
   cc @tqchen 


-- 
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 #13663: [TIR] Create Layout with specified axis dtype

[GitBox <gi...@apache.org>]

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

   <!---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, @junrushao, @quic-sanirudh <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] masahi merged pull request #13663: [TIR] Create Layout with specified axis dtype

[GitBox <gi...@apache.org>]

masahi merged PR #13663:
URL: https://github.com/apache/tvm/pull/13663


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