Skip to content

[TensorIR][M2a] Verification of cached flags#8114

Merged
comaniac merged 3 commits intoapache:mainfrom
junrushao:feature/2021-05-23/verify-cached-flags
May 25, 2021
Merged

[TensorIR][M2a] Verification of cached flags#8114
comaniac merged 3 commits intoapache:mainfrom
junrushao:feature/2021-05-23/verify-cached-flags

Conversation

@junrushao
Copy link
Copy Markdown
Member

This PR is part of the TensorIR upstreaming effort (#7527), stage M2a.

In this PR, we implemented cached flag verification, which will check the correctness of the following flags if the schedule class is set to debug:

  • is_affine_binding
  • region_cover
  • is_stage_pipeline

To enable the detection of the region cover property, we also bring in several integer set analysis along with this PR.

@junrushao junrushao changed the title [TensorIR][M2a] Verification of cached flags (#390) [TensorIR][M2a] Verification of cached flags May 23, 2021
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
@junrushao
Copy link
Copy Markdown
Member Author

CC: @tqchen @jroesch @comaniac @icemelon9 @zhiics

Comment thread src/arith/int_set.cc Outdated
Comment thread src/arith/int_set.cc Outdated
Comment thread src/arith/int_set.cc Outdated
Copy link
Copy Markdown
Member

@tqchen tqchen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, some minor comments on readability

Comment thread src/tir/schedule/analysis/analysis.cc
@tqchen tqchen self-assigned this May 24, 2021
Copy link
Copy Markdown
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Only nits and feel free to ignore.

Comment thread src/tir/schedule/analysis/verify.cc Outdated
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
@junrushao junrushao requested review from comaniac and tqchen May 25, 2021 00:01
@comaniac comaniac merged commit 47c8e47 into apache:main May 25, 2021
@comaniac
Copy link
Copy Markdown
Contributor

Thanks @junrushao1994 @tqchen

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Jun 17, 2021
* [TensorIR][M2a] Verification of cached flags

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

* Address comments

* Update src/tir/schedule/analysis/verify.cc

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Jun 17, 2021
* [TensorIR][M2a] Verification of cached flags

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

* Address comments

* Update src/tir/schedule/analysis/verify.cc

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants