Skip to content

Comments

[Feature] [Layout] Add layout of global buffer#53

Merged
wanghz18 merged 2 commits intotilelang_mesh_mainfrom
u/jiaqiguo/dram_layout
Feb 10, 2026
Merged

[Feature] [Layout] Add layout of global buffer#53
wanghz18 merged 2 commits intotilelang_mesh_mainfrom
u/jiaqiguo/dram_layout

Conversation

@JiaqiGuoSunlune
Copy link
Collaborator

Add MeshTensor into LayoutMap in layout_inference, so that they can be used easily for lowering

Copy link
Collaborator

@wanghz18 wanghz18 left a comment

Choose a reason for hiding this comment

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

整体没有问题,但是有一个小的concern。印象里有些地方会遍历layoutmap,然后针对[buffer, layout]做处理,如果现在layoutmap里新增了这些global的layout是否会导致报错,因为他的处理逻辑基本都是默认是shared或者local。不过目前在分支上简单跑了一些例子,没遇到问题。

@JiaqiGuoSunlune
Copy link
Collaborator Author

整体没有问题,但是有一个小的concern。印象里有些地方会遍历layoutmap,然后针对[buffer, layout]做处理,如果现在layoutmap里新增了这些global的layout是否会导致报错,因为他的处理逻辑基本都是默认是shared或者local。不过目前在分支上简单跑了一些例子,没遇到问题。

我查了一下,确实有很多地方遍历layoutmap,对layout的修改也没有作检查,所以我修改了一下,现在单独加了一个 global_layout_map,使用 LowerArgs.global_layout_map 来获取 global tensor layout

@wanghz18 wanghz18 merged commit e99bf09 into tilelang_mesh_main Feb 10, 2026
2 checks passed
@wanghz18 wanghz18 deleted the u/jiaqiguo/dram_layout branch February 10, 2026 06:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants