[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).

Reply via email to