Skip to content

[MetaSchedule][M4a] Rewrite-Cooperative-Fetch#10081

Merged
spectrometerHBH merged 1 commit into
apache:mainfrom
Hzfengsy:rewrite-cooperative-fetch
Jan 29, 2022
Merged

[MetaSchedule][M4a] Rewrite-Cooperative-Fetch#10081
spectrometerHBH merged 1 commit into
apache:mainfrom
Hzfengsy:rewrite-cooperative-fetch

Conversation

@Hzfengsy

Copy link
Copy Markdown
Member

It is one of the PostProc for mate-schedule. Please see the tracking issue #8473

Co-authored-by: Junru Shao junrushao1994@gmail.com
Co-authored-by: Xiyou Zhou xiyou@octoml.ai
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

cc @junrushao1994 @comaniac

@Hzfengsy Hzfengsy changed the title [MetaSchedule] postproc: rewrite_cooperative_fetch [MetaSchedule][M4a] Rewrite cooperative fetch Jan 27, 2022
@junrushao junrushao changed the title [MetaSchedule][M4a] Rewrite cooperative fetch [MetaSchedule][M4a] Rewrite-Cooperative-Fetch Jan 27, 2022
@junrushao

Copy link
Copy Markdown
Member

Hey please rebase to the latest main

@Hzfengsy Hzfengsy force-pushed the rewrite-cooperative-fetch branch from d90b791 to 4c3ba47 Compare January 28, 2022 06:44

@MasterJH5574 MasterJH5574 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM! Only have some nits. Thank you Siyuan!

Comment thread src/meta_schedule/postproc/rewrite_cooperative_fetch.cc Outdated
Comment thread src/meta_schedule/postproc/rewrite_cooperative_fetch.cc Outdated
Comment thread src/meta_schedule/postproc/rewrite_cooperative_fetch.cc Outdated
@Hzfengsy Hzfengsy force-pushed the rewrite-cooperative-fetch branch from 4c3ba47 to ce52b56 Compare January 29, 2022 05:53
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
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>
@Hzfengsy Hzfengsy force-pushed the rewrite-cooperative-fetch branch from ce52b56 to 70c014e Compare January 29, 2022 06:27
@spectrometerHBH spectrometerHBH merged commit 538347e into apache:main Jan 29, 2022
ylc pushed a commit to ylc/tvm that referenced this pull request Feb 16, 2022
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
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: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
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>
tir::ParseThreadBinding(sch, inst, "threadIdx.y")) {
thread_extent_y = new_thread_extent.value()->value;
} else if (Optional<tir::BlockRV> block_rv = tir::ParseAnnotate(sch, inst, &vector_lane)) {
ICHECK_NE(thread_extent_x, -1);

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

is this check necessary here? I am hitting this check with a simple matmul_fp16 example with rules and postprocs like in test_meta_schedule_tune_tir and I wonder if more details about this check can be elaborated here.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

it means threadIdx.x isn't bound in previous instructions, which isn't supposed to happen (and that's why it's an ICHECK instead of CHECK). Could you check the trace->insts and see why threadIdx.x doesn't exist previously

@Hzfengsy Hzfengsy deleted the rewrite-cooperative-fetch branch February 2, 2024 08:29
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.

5 participants