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 moderato via TVM Discuss <no...@discuss.tvm.ai> on 2020/05/25 20:38:31 UTC

[TVM Discuss] [Questions] Do pragmas cover all the operators/nested loops attached under the specified kernel_scope?


Hello! In the GPU conv tutorial I see the following two lines:

```python
s[output].pragma(kernel_scope, 'auto_unroll_max_step', cfg['auto_unroll_max_step'].val)
s[output].pragma(kernel_scope, 'unroll_explicit', cfg['unroll_explicit'].val)
```

I'm curious if these pragmas cover all the operators/nested loops, e.g. AA, WW in the tutorial, that are attached under `kernel_scope` which is the first `iter_var` of `output`? Generally, suppose I have a schedule looks like this: 

```
for A {
    for B {
    }
    for C {
    }
}
```

If I set the kernel_scope to be `for A`, does it take effect on `for B` and `for C`? If I want `for B` and `for C` to follow different pragmas, should I define two separate groups of tunable knobs for both of them?

Thanks in advance!





---
[Visit Topic](https://discuss.tvm.ai/t/do-pragmas-cover-all-the-operators-nested-loops-attached-under-the-specified-kernel-scope/6802/1) 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/5e3caf801d8f09d53e66eb5e3dbf8cc9463260d223382637ce63e03c6c791214).