[Apache TVM Discuss] [Questions] Optimizing a loop body expressed with tvm.extern

2021-03-07 Thread Teng HUANG via Apache TVM Discuss
May I ask, in this case, the only way to optimize the extern op is to do it by hand (with tir intrinsics) right ? For example like topi.sparse.csrmm implementation, no schedule configs could be associated with it. --- [Visit Topic](https://discuss.tvm.apache.org/t/optimizing-a-loop-body-e

[Apache TVM Discuss] [Questions] [Relay][topi/nn] Ways to get layout of each operator

2021-03-07 Thread xhuohai via Apache TVM Discuss
[quote="comaniac, post:2, topic:9310"] The way I can think of is just infer the layout manually from input tensor or the op with the layout attribute. [/quote] Thanks @comaniac, that's what I'm doing right now. --- [Visit Topic](https://discuss.tvm.apache.org/t/relay-topi-nn-ways-to-get-la