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

Posted to dev@tvm.apache.org by Leonard Mosescu via Apache TVM Discuss <no...@discuss.tvm.ai> on 2022/02/10 21:03:10 UTC

[Apache TVM Discuss] [Development] LaunchParamConfig documentation?


On a related note, can someone point me to the code responsible for calculating kernel launch parameters?





---
[Visit Topic](https://discuss.tvm.apache.org/t/launchparamconfig-documentation/11968/2) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/76636fd1865ef84037f27d918bc24ef660f9b1cd9fa5f11c7aadfa51f70c44a5).

[Apache TVM Discuss] [Development] LaunchParamConfig documentation?

Posted by masahi via Apache TVM Discuss <no...@discuss.tvm.ai>.


[quote="lemo, post:2, topic:11968"]
can someone point me to the code responsible for calculating kernel launch parameters
[/quote]

This is done by each op separately, most of them written in python, for example 
* https://github.com/apache/tvm/blob/bef7bf9b2b326488bc2bcc039710b2723023aaa0/tests/python/unittest/test_tir_ir_builder.py#L509-L512. 
* https://github.com/apache/tvm/blob/e082ef5f9db89d3f0ff20340717b9d7547a25eaf/python/tvm/topi/cuda/sort.py#L61-L69

Note that the launch param also includes dynamic shmem size. These params that are specified in python are retrieved in `split_host_device.cc` at https://github.com/apache/tvm/blob/main/src/tir/transforms/split_host_device.cc#L46





---
[Visit Topic](https://discuss.tvm.apache.org/t/launchparamconfig-documentation/11968/4) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/dad345a1f76ab0ffc79d92e0a6bb022e6a9884ba3e4294f67144d129acb46e22).