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

Posted to discuss-archive@tvm.apache.org by Ghostplant via TVM Discuss <no...@discuss.tvm.ai> on 2020/06/30 07:52:38 UTC

[TVM Discuss] [Questions] How to schedule fused ops?


Hi, can you show the `critical schedule statement` that makes the conv2d op fused into its following elementwise op, e.g. topi.nn.relu?

I found it should use `compute_at` but it seems to be a little hard to use that well.
```
B = conv2d(A)
C = topi.nn.relu(B)
..
schedule_for_conv2d(s[B], ..)
s[B].compute_at(s[C], C.op.axis[what is reasonable?])
```





---
[Visit Topic](https://discuss.tvm.ai/t/how-to-schedule-fused-ops/2522/9) 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/a4c97bef95ec8ce7f477d864461ff404db22a864b1b001be253063172e4c94fc).

[TVM Discuss] [Questions] How to schedule fused ops?

Posted by Wuwei Lin via TVM Discuss <no...@discuss.tvm.ai>.


https://github.com/apache/incubator-tvm/blob/master/topi/python/topi/cuda/conv2d.py#L46

https://github.com/apache/incubator-tvm/blob/master/topi/python/topi/cuda/conv2d_direct.py#L62





---
[Visit Topic](https://discuss.tvm.ai/t/how-to-schedule-fused-ops/2522/10) 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/5971ffd4d9a40b371790e7df1c1c3c2cff82fa78412b4c27d3c3ae0cf4852607).