triton-ascend

wutianyaotesting: Add select analysis test cases

文件	最后提交记录	最后更新时间
.github	[BUILD] Fix mac build CI (#7997) We were picking the wrong clang after image upgrade. Solution mostly from chatgpt	9 个月前
bin	Merge Triton-Ascend 62eb951f into release/3.5.x	1 个月前
cmake	feat: update llvm hash and upload fad3272.patch Co-authored-by: candyhong<1102229410@qq.com> # message auto-generated for no-merge-commit merge: !60 merge release/3.5.x-upgrade-candy-dev into release/3.5.x-upgrade feat: update llvm hash and upload fad3272.patch Created-by: candyhong Commit-by: candyhong Merged-by: zhuxuejie Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## checklist <!-- [x] 表示选中 --> - [ ] 是否通过本地IDE对代码进行静态检查 - [ ] 是否通过本地IDE对代码进行格式化处理 - [ ] 是否进行空指针校验 - [ ] 是否进行返回值校验 - [ ] 是否正确释放new/malloc申请的内存 - [ ] 是否充分考虑接口的异常场景 - [ ] 是否正确记录错误日志 See merge request: jeshd/triton-ascend!60	1 个月前
docker	fix(docs)Update CANN_TYPE to uniformly use a2 and a3; fix some expression issues in the documentation. Co-authored-by: wuyao51511<wuyao61@h-partners.com> Co-authored-by: xuweixin_<xuweixin2@huawei.com> # message auto-generated for no-merge-commit merge: !1634 merge fix_cann_type_docs into main fix(docs)Update CANN_TYPE to uniformly use a2 and a3; fix some expression issues in the documentation. Created-by: xuweixin_ Commit-by: xuweixin_;wuyao51511 Merged-by: ascend-robot Description: Update CANN_TYPE to uniformly use A2 and A3, no 910; fix some expression issues in the documentation. update en docs See merge request: Ascend/triton-ascend!1634	1 个月前
docs	fix(docs)Update CANN_TYPE to uniformly use a2 and a3; fix some expression issues in the documentation. Co-authored-by: wuyao51511<wuyao61@h-partners.com> Co-authored-by: xuweixin_<xuweixin2@huawei.com> # message auto-generated for no-merge-commit merge: !1634 merge fix_cann_type_docs into main fix(docs)Update CANN_TYPE to uniformly use a2 and a3; fix some expression issues in the documentation. Created-by: xuweixin_ Commit-by: xuweixin_;wuyao51511 Merged-by: ascend-robot Description: Update CANN_TYPE to uniformly use A2 and A3, no 910; fix some expression issues in the documentation. update en docs See merge request: Ascend/triton-ascend!1634	1 个月前
include	Merge Triton-Ascend 425236de into release/3.5.x	2 个月前
lib	Merge Triton-Ascend 425236de into release/3.5.x	2 个月前
python	fix: raise error for unsupported allow_tf32 in dot_op Co-authored-by: ZhangAiqiang<zhangaiqiang1@huawei.com> # message auto-generated for no-merge-commit merge: !1643 merge tf32_ban_main into main fix: raise error for unsupported allow_tf32 in dot_op Created-by: zaq15csdn Commit-by: ZhangAiqiang Merged-by: ascend-robot Description: <!--- The core Triton is a small number of people, and we receive many PRs (thank you!). To help us review your code more quickly, if you are a new contributor (less than 3 PRs merged) we ask that you complete the following tasks and include the filled-out checklist in your PR description. Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> # New contributor declaration - [x] I am not making a trivial change, such as fixing a typo in a comment. - [x] I have written a PR description following these [rules](https://cbea.ms/git-commit/#why-not-how). - [ ] I have run `pre-commit run --from-ref origin/main --to-ref HEAD`. - Select one of the following. - [x] I have added tests. - `/test` for `lit` tests - `/unittest` for C++ tests - `/python/test` for end-to-end tests - [ ] This PR does not need a test because `FILL THIS IN`. - Select one of the following. - [x] I have not added any `lit` tests. - [ ] The `lit` tests I have added follow these [best practices](https://mlir.llvm.org/getting_started/TestingGuide/#filecheck-best-practices), including the "tests should be minimal" section. (Usually running Python code and using the instructions it generates is not minimal.) See merge request: Ascend/triton-ascend!1643	1 个月前
scripts	[Build] change default llvm build to release (#7872)	9 个月前
test	Add `llc` to the lit tool configuration (#7992) The triton lit tests leverage several llvm & mlir tools such as opt, mlir-translate and llc. These are generally configured to come directly from the llvm build that triton is using, but llc is not. This means that right now the llc version on the system will be used instead and if it doesn't match the rest of the tools, we can end up with failures. An example failure is using an older version than the llvm build and missing features (such as `ptx83`). To fix this issue, configure lit to use the `llc` directly from the llvm build.	9 个月前
third_party	testing: Add select analysis test cases	1 个月前
unittest	[Swizzling] Disallow asymmetric vectorisation within generic swizzling (#7833) Before, we would generate bank conflicts in, say, the load, as we were too eager to use `st.shared.b32.vn`. Now in these cases, we do not use vectorisation on the stores, but we do not create bank conflicts on the load either. We leave a comment as to how we could do better, but it'd be a bit of effort for rather modest wins (if any). Fixes https://github.com/triton-lang/triton/issues/7815	9 个月前
utils	Fix typos discovered by codespell (#5533) ``` codespell --ignore-words-list=ans,conver,destop,dout,interm,latops,nd,olt,repid,subtile,valu \ --skip="./third_party/,.mlir,.pdf" ``` https://pypi.org/project/codespell `conver` should be removed from the `ignore-words-list` and the following line should be cleaned up: https://github.com/triton-lang/triton/blob/25732f0a063f025a4eb6f01a2e0939ef43a8afc7/include/triton/Dialect/TritonGPU/Transforms/Passes.td#L122 Should codespell be added to pre-commit and/or GitHub Actions? <!--- The core Triton is a small number of people, and we receive many PRs (thank you!). To help us review your code more quickly, if you are a new contributor (less than 3 PRs merged) we ask that you complete the following tasks and include the filled-out checklist in your PR description.* Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> # New contributor declaration - [x] I am making a trivial change, such as fixing a typo in a comment. - [ ] I have written a PR description following these [rules](https://cbea.ms/git-commit/#why-not-how). - [ ] I have run `pre-commit run --from-ref origin/main --to-ref HEAD`. - Select one of the following. - [ ] I have added tests. - `/test` for `lit` tests - `/unittest` for C++ tests - `/python/test` for end-to-end tests - [x] This PR does not add a test but this could be easily added to pre-commit and/or GitHub Actions if desired. - Select one of the following. - [ ] I have not added any `lit` tests. - [ ] The `lit` tests I have added follow these [best practices](https://mlir.llvm.org/getting_started/TestingGuide/#filecheck-best-practices), including the "tests should be minimal" section. (Usually running Python code and using the instructions it generates is not minimal.) --------- Co-authored-by: Keren Zhou <robinho364@gmail.com>	1 年前
.clang-format	refactor(docs): Docs update	4 个月前
.editorconfig	[CI][TEST] update `pre-commit` hooks and use `pre-commit` for style tests in CI (#1409) Ref issue: - #1408 Changes: - Add `.editorconfig` - Add `pre-commit-hooks`: ```yaml - repo: https://github.com/pre-commit/pre-commit-hooks rev: v4.4.0 hooks: - id: check-symlinks - id: destroyed-symlinks - id: trailing-whitespace - id: end-of-file-fixer - id: check-yaml - id: check-toml - id: check-ast - id: check-added-large-files - id: check-merge-conflict - id: check-executables-have-shebangs - id: check-shebang-scripts-are-executable - id: detect-private-key - id: debug-statements ``` - Add `flake8` to `pre-commit` config and add `.flake8` file - Use `pre-commit` for style tests in CI - Run `pre-commit` and fix existing violations: - fix trailing spaces - fix end-of-files - fix mod file mode with `chmod -x` - run `autopep8` on existing code - fix `flake8` violations	3 年前
.git-blame-ignore-revs	Add .git-blame-ignore-revs. (#3117) I've been making some big NFC changes, and these can clog up `git blame`. Add a file that lists "uninteresting" revisions. You can (optionally!) instruct git to ignore the revs in here during blame. I got this list of commits by looking through the list of Triton commits that touch 30+ files. ChatGPT wrote the following bash script for me. git log --pretty=format:"%H" --name-only \| \ awk -v N="30" ' /^[0-9a-f]{40}$/ { if (count > N) print hash; hash=$0; count=0; } NF { count++ } END { if (count > N) print hash } ' \| xargs git show --stat	2 年前
.gitignore	Merge Triton 5389ed7 into release/3.4.x first.	3 个月前
.gitmodules	feat(compile): add npu-ir to submodule Co-authored-by: luobaiqing<luobaiqing1@huawei.com> # message auto-generated for no-merge-commit merge: !1085 merge submodule into main feat(compile): add npu-ir to submodule Created-by: luobaiqing Commit-by: luobaiqing Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## checklist <!-- [x] 表示选中 --> - [ ] 是否通过本地IDE对代码进行静态检查 - [ ] 是否通过本地IDE对代码进行格式化处理 - [ ] 是否进行空指针校验 - [ ] 是否进行返回值校验 - [ ] 是否正确释放new/malloc申请的内存 - [ ] 是否充分考虑接口的异常场景 - [ ] 是否正确记录错误日志 See merge request: Ascend/triton-ascend!1085	4 个月前
.pre-commit-config.yaml	[RFC] Enable mypy typechecking on subset of files in pre-commit (#6704) In https://github.com/triton-lang/triton/pull/6497 and https://github.com/triton-lang/triton/pull/6467 I took care to ensure the relevant modules typechecked successfully with mypy. Now I want to make sure they stay this way, so I'm proposing including `mypy` in the standard pre-commit flow. This way we can iteratively make different files typecheck without fear of losing progress. Example UX when creating type error: ``` > git diff --cached diff --git a/python/triton/knobs.py b/python/triton/knobs.py index dd291513d..eab3cca17 100644 --- a/python/triton/knobs.py +++ b/python/triton/knobs.py @@ -106,7 +106,7 @@ class env_int(env_base[int, int]): def from_env(self, val: str) -> int: try: - return int(val) + return str(int(val)) except ValueError as exc: raise RuntimeError(f"Unable to use {self.key}={val}: expected int") from exc > git commit -m tst check for broken symlinks............................(no files to check)Skipped detect destroyed symlinks................................................Passed trim trailing whitespace.................................................Passed fix end of files.........................................................Passed check yaml...........................................(no files to check)Skipped check toml...........................................(no files to check)Skipped check python ast.........................................................Passed check for added large files..............................................Passed check for merge conflicts................................................Passed check that executables have shebangs.................(no files to check)Skipped check that scripts with shebangs are executable..........................Passed detect private key.......................................................Passed debug statements (python)................................................Passed ruff.....................................................................Passed yapf.....................................................................Passed clang-format.........................................(no files to check)Skipped mypy.....................................................................Failed - hook id: mypy - exit code: 1 python/triton/knobs.py:109: error: Incompatible return value type (got "str", expected "int") [return-value] Found 1 error in 1 file (checked 3 source files) Expand YAML anchors..................................(no files to check)Skipped ``` And example UX when changing file that isn't typechecked (even though the file has type errors): ``` > git diff --cached diff --git a/python/triton/language/core.py b/python/triton/language/core.py index 397b87560..9339cde9b 100644 --- a/python/triton/language/core.py +++ b/python/triton/language/core.py @@ -97,6 +97,7 @@ def _unwrap_iterable(x): # right answer for whether the class constexpr defines __iter__, and # abc.Iterable doesn't work (at least not without some metaclass magic). try: + _a: str = 5 iter(x[0]) return x[0] except TypeError: > git commit -m tst check for broken symlinks............................(no files to check)Skipped detect destroyed symlinks................................................Passed trim trailing whitespace.................................................Passed fix end of files.........................................................Passed check yaml...........................................(no files to check)Skipped check toml...........................................(no files to check)Skipped check python ast.........................................................Passed check for added large files..............................................Passed check for merge conflicts................................................Passed check that executables have shebangs.................(no files to check)Skipped check that scripts with shebangs are executable..........................Passed detect private key.......................................................Passed debug statements (python)................................................Passed ruff.....................................................................Passed yapf.....................................................................Passed clang-format.........................................(no files to check)Skipped mypy.....................................................................Passed Expand YAML anchors..................................(no files to check)Skipped [danzimm/mypycheck 9709ff131] tst 1 file changed, 1 insertion(+) ``` N.B. In order to avoid duplicate inclusion/exclusion lists in `pyproject.toml` and `.pre-commit-config.yaml` I've disabled passing file names from pre-commit (since the CLI files override any lists in `pyproject.toml`), since if we only specify the exlusions in `pre-commit` running `mypy` directly wouldn't pick up that configuration.	1 年前
.readthedocs.yaml	fix(docs): modify profiling Co-authored-by: HallCin<qinhao61@h-partners.com> # message auto-generated for no-merge-commit merge: !1241 merge br_main into main fix(docs): modify profiling Created-by: HallCin Commit-by: HallCin Merged-by: ascend-robot Description: The core Triton is a small number of people, and we receive many PRs (thank you!). To help us review your code more quickly, if you are a new contributor (less than 3 PRs merged) we ask that you complete the following tasks and include the filled-out checklist in your PR description. Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. - [ ] I am not making a trivial change, such as fixing a typo in a comment. - [ ] I have written a PR description following these [rules](https://cbea.ms/git-commit/#why-not-how). - [ ] I have run `pre-commit run --from-ref origin/main --to-ref HEAD`. - Select one of the following. - [ ] I have added tests. - `/test` for `lit` tests - `/unittest` for C++ tests - `/python/test` for end-to-end tests - [ ] This PR does not need a test because `FILL THIS IN`. - Select one of the following. - [ ] I have not added any `lit` tests. - [ ] The `lit` tests I have added follow these [best practices](https://mlir.llvm.org/getting_started/TestingGuide/#filecheck-best-practices), including the "tests should be minimal" section. (Usually running Python code and using the instructions it generates is not minimal.) See merge request: Ascend/triton-ascend!1241	3 个月前
CMakeLists.txt	Merge Triton-Ascend 62eb951f into release/3.5.x	1 个月前
CONTRIBUTING.md	fix doctools probleam	1 个月前
CONTRIBUTING_zh.md	fix doctools probleam	1 个月前
LICENSE	feat(docs): migrate documents from master branch Co-authored-by: zhang-chunli01<zhangchunli19@huawei.com> # message auto-generated for no-merge-commit merge: !1074 merge migrate-docs into main feat(docs): migrate documents from master branch Created-by: zhang-chunli01 Commit-by: zhang-chunli01 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## checklist <!-- [x] 表示选中 --> - [ ] 是否通过本地IDE对代码进行静态检查 - [ ] 是否通过本地IDE对代码进行格式化处理 - [ ] 是否进行空指针校验 - [ ] 是否进行返回值校验 - [ ] 是否正确释放new/malloc申请的内存 - [ ] 是否充分考虑接口的异常场景 - [ ] 是否正确记录错误日志 See merge request: Ascend/triton-ascend!1074	4 个月前
MANIFEST.in	Use symlinks for external plugins to fix TRITON_PLUGIN_DIRS (#6627) Disable the new backend installing logic for external plugins, since `package_dir` does not accept absolute paths. Instead, use a hybrid approach where in-tree backends are installed using the new logic and external backends are symlinked. This implies that source distributions cannot be created when using external plugins. Fixes #6612 ------ <!--- The core Triton is a small number of people, and we receive many PRs (thank you!). To help us review your code more quickly, if you are a new contributor (less than 3 PRs merged) we ask that you complete the following tasks and include the filled-out checklist in your PR description. Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> # New contributor declaration - [x] I am not making a trivial change, such as fixing a typo in a comment. - [x] I have written a PR description following these [rules](https://cbea.ms/git-commit/#why-not-how). - [x] I have run `pre-commit run --from-ref origin/main --to-ref HEAD`. - Select one of the following. - [ ] I have added tests. - `/test` for `lit` tests - `/unittest` for C++ tests - `/python/test` for end-to-end tests - [x] This PR does not need a test because it touches build system only. - Select one of the following. - [x] I have not added any `lit` tests. - [ ] The `lit` tests I have added follow these [best practices](https://mlir.llvm.org/getting_started/TestingGuide/#filecheck-best-practices), including the "tests should be minimal" section. (Usually running Python code and using the instructions it generates is not minimal.)	1 年前
Makefile	build: update LLVM prebuilt package generation for AlmaLinux and improve test coverage Co-authored-by: wangzhanpeng5<wangzhanpeng5@huawei.com> # message auto-generated for no-merge-commit merge: !1633 merge 0416_main_coverage into main build: update LLVM prebuilt package generation for AlmaLinux and improve test coverage Created-by: wangzhanpeng5 Commit-by: wangzhanpeng5 Merged-by: ascend-robot Description: ### 1.Update LLVM Prebuilt Package Generation for AlmaLinux - Enhanced the Makefile to properly support building and packaging a precompiled LLVM tarball on AlmaLinux. - Added explicit handling of Python dependencies (pybind11, nanobind >=2.4, and MLIR requirements) and configured CMake to correctly locate Python and binding libraries in virtual environments. ### 2.Improve Test Case Coverage See merge request: Ascend/triton-ascend!1633	1 个月前
README.md	fix doctools probleam	1 个月前
README_zh.md	fix doctools probleam	1 个月前
RELEASE.md	Add RELEASE.md (#5896) Specify Release Compatibility Matrix, Release Cadence and Release Cherry-Pick Criteria --------- Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com> Co-authored-by: Keren Zhou <robinho364@gmail.com>	1 年前
SECURITYNOTE.md	fix doctools probleam	1 个月前
SECURITYNOTE_zh.md	fix doctools probleam	1 个月前
pyproject.toml	[gluon] Remove programmatic imports for the sake of mypy (#7952) Follow up to #7923	9 个月前
requirements.txt	fix(docs): Improve install-deploy-readme jump configuration, add Docker parameter description and format adaptation Co-authored-by: xuweixin<xuweixin2@huawei.com> # message auto-generated for no-merge-commit merge: !1244 merge dev_DockerCannOptionAndInstallGuide into main fix(docs): Improve install-deploy-readme jump configuration, add Docker parameter description and format adaptation Created-by: xuweixin_ Commit-by: xuweixin Merged-by: ascend-robot Description: 1、Add CANN version selection for Docker-based installation; 2、update installation and usage instructions, adding quick installation and Docker-based installation methods; 3、correct the numpy version in requirements to 1.26.4 See merge request: Ascend/triton-ascend!1244	3 个月前
requirements_dev.txt	fix: add nanobind>=2.4 in requirements_dev.txt Co-authored-by: candyhong<1102229410@qq.com> # message auto-generated for no-merge-commit merge: !65 merge release/3.5.x-upgrade-candy-dev into release/3.5.x-upgrade fix: add nanobind>=2.4 in requirements_dev.txt Created-by: candyhong Commit-by: candyhong Merged-by: candyhong Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## checklist <!-- [x] 表示选中 --> - [ ] 是否通过本地IDE对代码进行静态检查 - [ ] 是否通过本地IDE对代码进行格式化处理 - [ ] 是否进行空指针校验 - [ ] 是否进行返回值校验 - [ ] 是否正确释放new/malloc申请的内存 - [ ] 是否充分考虑接口的异常场景 - [ ] 是否正确记录错误日志 See merge request: jeshd/triton-ascend!65	1 个月前
safe_compile.cmake	build(safe):Add safe compile opitions for libtriton.so libentryC.so Co-authored-by: wangzhanpeng5<wangzhanpeng5@huawei.com> # message auto-generated for no-merge-commit merge: !1535 merge 0402_main_safe_compile into main build(safe):Add safe compile opitions for libtriton.so libentryC.so Created-by: wangzhanpeng5 Commit-by: wangzhanpeng5 Merged-by: ascend-robot Description: To comply with the secure compilation option specifications`BIND_NOW`, `Strip`. Add safe compilation options for `libtriton.so` and `libentryC.so`. See merge request: Ascend/triton-ascend!1535	2 个月前
setup.py	fix(packaging): avoid triton overwriting triton-ascend after installation	1 个月前
version.txt	fix: create reinterpret_cast with dynamic and static parameters.	2 个月前

Triton-Ascend

项目简介与价值主张

Triton-Ascend是面向昇腾平台构建的Triton编译框架，旨在让Triton代码能够在昇腾硬件上高效运行。

核心价值说明

Triton是近几年来受到开发者青睐的Python化编译框架。开发者仅需关注Tile/Block的切分方式以及基于Tile/Block的运算逻辑，编译器将在Triton代码的编译过程中结合底层硬件特点自动完成内存分配、数据搬运、数据计算、流水并行等，因此，算子的开发难度大幅降低、开发效率显著提升。 Triton-Ascend将Triton编译栈适配到华为昇腾NPU上，在Triton的基础上提供一系列针对性的优化，使Triton代码能够编译后在昇腾硬件上高效运行。目前，Triton-Ascend仍在持续完善中，我们将不断提升Triton Python API完备度、数据类型支持度、访存方式灵活性等，并持续优化编译器的自动优化能力，提升Triton-Ascend整体的功能与性能泛化性。

昇腾生态定位

Triton-Ascend编译框架打通了Triton与昇腾硬件之间的壁垒，使熟悉Triton框架的开发者可以更有效率地使用昇腾NPU。它通过提供通用、高效的算子开发范式，为昇腾软件栈补齐了敏捷开发的关键一环，极大丰富了昇腾的算子库和上层应用生态。

里程碑	重要特性更新情况	状态
2026.04	Triton-Ascend 3.2.0 post版本上线
2026.01.20	Triton-Ascend 3.2.0 正式版本上线	✅
2025.11.14	Triton-Ascend 3.2.0rc4预发布版本上线：扩展 tt.fp_to_fp 接口，新增对 FP8的类型转换支持新增 scatter_ub_to_out 接口，支持从UB到GM的高效数据分散操作	✅
2025.09.30	完善Scan/Sort类Triton Python API，支持非连续访存，完成vLLM、sglang开源仓中重点Triton算子适配	✅
2025.09.19	支持Triton-Ascend nightly包提取	✅
2025.08.15	完善Atomic类Triton Python API支持，完成Flaggems开源仓重点Triton算子适配，提供Matmul等简单算子高性能实现参考用例	✅
2025.06.30	支持85% Triton Python API，支持连续访存，覆盖基本使用场景需求	✅
2025.05.20	Triton-Ascend开源，Gitcode代码仓Alive！	✅

性能

GroupGEMM 算子性能

选取GroupGEMM算子作为示例。展示Triton-Ascend与AscendC的性能对比。

Y轴为加速比(Speedup= AscendC_Duration_Time / Triton_Duration_Time)
使用硬件为Ascend 950系列
算子性能优化请参考算子调优指南：

支持范围

硬件支持

Triton-Ascend 在昇腾 AI 产品支持使用，具体型号如下：

产品系列	产品型号
Atlas A3 训练系列产品	Atlas 800T A3 超节点服务器 Atlas 900 A3 SuperPoD 超节点 A200T A3 Box8 超节点服务器
Atlas A3 推理系列产品	Atlas 800I A3 超节点服务器
Atlas A2 训练系列产品	Atlas 800T A2 训练服务器 Atlas 900 A2 PoD 集群基础单元 Atlas 200T A2 Box16 异构子框
Atlas A2 推理系列产品	Atlas 800I A2 推理服务器 Atlas 300I A2 推理卡 A200I A2 Box 异构组件