pku-liang / FlexTensor

Automatic Schedule Exploration and Optimization Framework for Tensor Computations

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to encode and schedule for "reorder"?

yzhang93 opened this issue Β· comments

Hello,

Thanks for the amazing work and codes.

I looked into the paper and codes but was confused about how you encode "reorder" schedule in the space. I have some questions.

  1. In the codes I found
    reorder_space = generate_reorder_space(groups) and groups = 3 by default
    Shall I change groups equal to rlevel*slevel?

  2. The configs it generated is just one dimensional, and does not match the description in the paper, i.e.,
    β€œreorder”: 𝑖1, 𝑗1, 𝑖2, 𝑗2, π‘˜1, 𝑖3, π‘˜2, 𝑗3, π‘˜3, 𝑖4, π‘˜4, 𝑗4 in Figure 3

Could you please explain these? Thanks.