[Auto TVM] How to specify particular axis to vectorize when reorder is also employed?

Questions
1

Hey guys, I want to utilize the auto-tuning capability of autotvm. Specifically, I try to apply tiling, loop reorder and vectorization on axes. I provide a list of candidates for reorder and I want to vectorize the innermost axis for each particular reorder candidate. As the innermost axis could be different for different reorder candidates, I think specifying the vectorization axis through its name seems inappropriate.

define_annotate ( name , axes , policy , **kwargs ) Does there are convenient ways to achieve this?

2

You can use cfg[name].perm[-1] to subscript the candidate list to get the innermost axis after reorder.
e.g https://github.com/dmlc/tvm/blob/a706ad16f83d810f4a6268cf604cece79c2c4791/topi/python/topi/cuda/conv2d_int8.py#L259