Support ptr plus scalar PTODSL JIT entries#4

Closed

jimmychou0 wants to merge 25 commits into

Zhendong404:pto-dsl-implfrom

jimmychou0:codex/issue417-ptr-entry

Commits on May 26, 2026

quick install script on top of MLIR docker image

learning-chip
authored and
Zhendong404
committed
add reference result for top->vop expansion

learning-chip
authored and
Zhendong404
committed
low-level python binding example to generate vpto IR

learning-chip
authored and
Zhendong404
committed
initial prototype of high-level dsl builder api

learning-chip
authored and
Zhendong404
committed
initial prototype of softmax IR builder

learning-chip
authored and
Zhendong404
committed
script to check IR equal

learning-chip
authored and
Zhendong404
committed
avoid raw MLIR `Type.parse`

learning-chip
authored and
Zhendong404
committed
more Pythonic builder style suggestions

learning-chip
authored and
Zhendong404
committed
major refactor of dsl syntax and impl

learning-chip
authored and
Zhendong404
committed
[vpto] Add ptodsl tracing POC
Zhendong404
committed
[vpto] Allow structured loops without vecscope
Zhendong404
committed
Add user guides
Zhendong404
committed
Add a flash attention demo
Zhendong404
committed
Completed the first version of PTODSL user guide
Zhendong404
committed
Complete the mlir text emission of the FA demo
Zhendong404
committed
pip install ptoas

castigli
authored and
Zhendong404
committed
use pip install in CI (#385)

castigli
authored and
Zhendong404
committed
feature(ptodsl): align ptodsl implementation with user guide
Zhendong404
committed
chore(ptodsl): normalize docs test headers
Zhendong404
committed
python builder to reproduce tilelang_st/tadd.pto

learning-chip
authored and
Zhendong404
committed
Switch to new kernel surface
Zhendong404
committed
Clean up the pending docs-as-test in the user guide
Zhendong404
committed
Clarify the pto.jit kernel signature
Zhendong404
committed
Refine the online softmax demo
Zhendong404
committed

Commits on May 27, 2026

Support ptr plus scalar JIT entries
jimmychou0
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Support ptr plus scalar PTODSL JIT entries#4

Support ptr plus scalar PTODSL JIT entries#4
jimmychou0 wants to merge 25 commits into
Zhendong404:pto-dsl-implfrom
jimmychou0:codex/issue417-ptr-entry

Commits on May 26, 2026

quick install script on top of MLIR docker image

add reference result for top->vop expansion

low-level python binding example to generate vpto IR

initial prototype of high-level dsl builder api

initial prototype of softmax IR builder

script to check IR equal

avoid raw MLIR `Type.parse`

more Pythonic builder style suggestions

major refactor of dsl syntax and impl

[vpto] Add ptodsl tracing POC

[vpto] Allow structured loops without vecscope

Add user guides

Add a flash attention demo

Completed the first version of PTODSL user guide

Complete the mlir text emission of the FA demo

pip install ptoas

use pip install in CI (#385)

feature(ptodsl): align ptodsl implementation with user guide

chore(ptodsl): normalize docs test headers

python builder to reproduce tilelang_st/tadd.pto

Switch to new kernel surface

Clean up the pending docs-as-test in the user guide

Clarify the pto.jit kernel signature

Refine the online softmax demo

Commits on May 27, 2026

Support ptr plus scalar JIT entries