Skip to content

Comments

[Example] Built-in with region & [Feature Request] [Pass] Copy To DMA#61

Merged
wanghz18 merged 6 commits intotilelang_mesh_mainfrom
u/jiaqiguo/builtin_with_region
Feb 11, 2026
Merged

[Example] Built-in with region & [Feature Request] [Pass] Copy To DMA#61
wanghz18 merged 6 commits intotilelang_mesh_mainfrom
u/jiaqiguo/builtin_with_region

Conversation

@JiaqiGuoSunlune
Copy link
Collaborator

Create an example to demonstrate how to lower TileOp into an intrinsic that takes T.region as arguments.

@wanghz18
Copy link
Collaborator

wanghz18 commented Feb 11, 2026

缺了一处改动在tilelang/language/copy.py中,见于Copy To DMA的pr中

@wanghz18 wanghz18 linked an issue Feb 11, 2026 that may be closed by this pull request
1 task
@wanghz18
Copy link
Collaborator

wanghz18 commented Feb 11, 2026

在这个Pull Request中,支持了T.region作为intrinsic的参数,因此DMA Copy和GEMM lower到intrinsic后的参数列表都可以大大简化。本PR中手动merge了DMA Copy的PR,因此原有DMA Copy的PR #60 可以关闭。

DMA Copy: Add InferLayout and Lower for dma_copy.

  1. Infer blockwise zz layout for copy node by default.
  2. Lower copy to dma_copy intrinsic: dma_copy(src_region, dst_region).

@wanghz18 wanghz18 changed the title [Example] Built-in with region [Example] Built-in with region & [Feature Request] [Pass] Copy To DMA Feb 11, 2026
@wanghz18 wanghz18 merged commit 5369162 into tilelang_mesh_main Feb 11, 2026
2 checks passed
@wanghz18 wanghz18 deleted the u/jiaqiguo/builtin_with_region branch February 11, 2026 09:23
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.

[Feature Request] [Pass] Copy To DMA

2 participants