Skip to content

[Merge] Put EDSL Code into triton_v3.5.x#272

Merged
sunnycase merged 21 commits intotriton_v3.5.xfrom
triton_v3.5_edsl_x
Jan 8, 2026
Merged

[Merge] Put EDSL Code into triton_v3.5.x#272
sunnycase merged 21 commits intotriton_v3.5.xfrom
triton_v3.5_edsl_x

Conversation

@sgjzfzzf
Copy link
Collaborator

@sgjzfzzf sgjzfzzf commented Jan 7, 2026

This PR resolves #271. It merges the commits of EDSL-related parts into triton_v3.5.x.

sgjzfzzf and others added 14 commits January 7, 2026 10:50
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Resolves #211.

* update setup to support llvm from wheel

* allow attempts mul times in one python progress

* delete redundant code

* delete redundant code _ 2

* delete redundant code_3
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
@CLAassistant
Copy link

CLAassistant commented Jan 7, 2026

CLA assistant check
All committers have signed the CLA.

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
@sgjzfzzf sgjzfzzf force-pushed the triton_v3.5_edsl_x branch from 52f42d1 to b9271a8 Compare January 8, 2026 02:29
@sgjzfzzf sgjzfzzf marked this pull request as ready for review January 8, 2026 02:30
Copy link
Collaborator

@zhzhcookie zhzhcookie left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Collaborator

@Galaxy1458 Galaxy1458 left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Collaborator

@sunnycase sunnycase left a comment

Choose a reason for hiding this comment

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

LGTM

@sunnycase sunnycase merged commit c056824 into triton_v3.5.x Jan 8, 2026
3 checks passed
@sunnycase sunnycase deleted the triton_v3.5_edsl_x branch January 8, 2026 06:37
zhzhcookie pushed a commit that referenced this pull request Mar 13, 2026
* Setup the framework of edsl

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* support scf.for loop

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* support softmax

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* support matmul

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* modified:   ../triton/experimental/flagtree/edsl/runtime.py (#216)

* add topk example skeleton

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* implement topk with edsl

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* replace while-loop-inside part with edsl

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* fix CMake Dependence bug (#247)

* [FEAT]enable use wheel as external LLVM (#248)

Resolves #211.

* update setup to support llvm from wheel

* allow attempts mul times in one python progress

* delete redundant code

* delete redundant code _ 2

* delete redundant code_3

* [BACKEND] Solve addresspace issue (#251)

resolve #209

* Reverse order of dimensions for memory descriptor

* Update typing.py

* support llvm.ptr as edsl input type

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* move the examples to tutorials

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* put edsl into tle

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* remove unused extension flag

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* tag modifications as tle raw

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

* Refactor: Move LLVM wheel logic to setup_helper.py

* Apply code-format changes

---------

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
Co-authored-by: Phoenixtree <sunwenjia04@163.com>
Co-authored-by: starrryz <760668919@qq.com>
Co-authored-by: flagtree-bot <flagtree_ai@163.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.

8 participants