It is actually possible to get multiple versions of function compiled together. 
Just need a bit of extra effort to do so. 

One way to achieve that now is to call tvm.lower(instead of buld) to get 
List[LoweredFuncs] for each of the function you care about(give them different 
names), then concat these lists together to get one list, then feed it to 
tvm.build to get you a single runtime module which which will contain all the 
functions you need. Then we can call mod.export_library to export a single so 
file that contains all the function variants. This is the approach we use to 
export multiple functions in a neural networks into a single shared library

Also we might have another way(by importing other modules into a single one) 
which @FrozenGene recently enabled.





---
[Visit 
Topic](https://discuss.tvm.ai/t/optimizing-matrix-multiplication-for-gpu/4212/21)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.ai/email/unsubscribe/2116b6406a5483635761368b84b1546a3b00a9aab3265df180ce18c112f8b5c3).

Reply via email to