Commit Graph

19 Commits

Author SHA1 Message Date
Rohit Santhanam
50e6329d01 Fix LIT tests. 2023-02-18 09:32:42 +00:00
Rohit Santhanam
841784d1e3 Merge remote-tracking branch 'upstream/main' into upgrade_triton_mlir_rocm_to_llvm_head 2023-02-18 09:25:20 +00:00
Alexander Efimov
fce08c2ebb [Test][LIT] Fix tritongpu_to_hsaco.mlir test
This PR fixes tritongpu_to_hsaco.mlir test,
plus fixes minor issue with tritongpu_to_amdgcn test
2023-02-16 20:15:52 +01:00
Christian Sigg
fc7a8e3581 Rebase Triton to LLVM-15. (#1070)
This PR rebases Triton from LLVM-14 to LLVM-15. Most changes are
mechanical, except for the analysis framework changes.
2023-02-16 06:40:53 -08:00
Daniil Fukalov
ce23494cbf [Test] Enable triton-translate build.
And use it in tests instead of python module triton.tools.aot.
2023-01-13 20:36:16 +01:00
Rohit Santhanam
7b7ddb7a59 Merge branch 'triton-mlir-IFU' into merge_IFU_to_triton_mlir 2023-01-03 23:37:11 +00:00
Michael Melesse
41578a63d2 Merge remote-tracking branch 'upstream/triton-mlir' into triton-mlir-IFU 2022-12-21 12:53:03 -06:00
Philippe Tillet
20100a7254 Merge triton-mlir branch - Complete rewrite of the backend from scratch (#1004)
This PR merges the `triton-mlir` branch, in which we have been quietly
rewriting the Triton backend from scratch to increase maintainability,
stability and ultimately performance. Changes to the runtime are
minimal, and this new version aims to remain backward-compatible with
the previous commit. The legacy backend is now officially deprecated,
but can still be accessed via the `legacy-backend` tag.

Co-authored-by: Keren Zhou <kerenzhou@openai.com>
Co-authored-by: Yan Chunwei <yanchunwei@outlook.com>
Co-authored-by: goostavz <109190422+goostavz@users.noreply.github.com>
Co-authored-by: Shintaro Iwasaki <siwasaki@fb.com>
Co-authored-by: Yan Da <dyanab@connect.ust.hk>
Co-authored-by: Jun Yang <yangjunpro@gmail.com>
Co-authored-by: Ian Bearman <ianb@microsoft.com>
Co-authored-by: Jason Ansel <jansel@jansel.net>
Co-authored-by: Qingyi Liu <qingyil@nvidia.com>
Co-authored-by: ben-zhang-609 <110140741+ben-zhang-609@users.noreply.github.com>
Co-authored-by: Chenggang Zhao <lyricz@yeah.net>
Co-authored-by: ben-zhang-609 <benzh609@gmail.com>
Co-authored-by: dongdongl <dongdongl@nvidia.com>
2022-12-21 01:30:50 -08:00
dfukalov
834c46ab44 [Triton-MLIR][HSACO] Added HSACO generation.
The change is based on Rohit' commit 9bf807f310

Used `AMDGCN_HSACO_DUMP_PATH=<dumpfile>` notion to save the copy of temp
generated HSACO file so test is able to llvm-readobj that dump and check
its content.
2022-11-28 23:34:56 +01:00
dfukalov
a6ae6dee9e [Triton-MLIR][Tests] Added trivial test for amdgcn target.
Additionally fixed typo in test for llvm-ir target.
2022-11-24 18:13:33 +01:00
ben-zhang-609
07786dc932 [Triton-MLIR] Add compute capability (#902)
add compute capability from python frontend to backend.

Co-authored-by: Keren Zhou <kerenzhou@openai.com>
2022-11-22 11:08:23 -08:00
Rohit Santhanam
b9e7634356 Merge commit 'e517b58d59ba96357d042d8fa5819a690d00d749' into IFU_upstream_commit_e517b58d59ba96357d042d8fa5819a690d00d749 2022-11-16 12:57:19 +00:00
Rohit Santhanam
ad7f53ab15 Merge commit 'e517b58d59ba96357d042d8fa5819a690d00d749' into triton-mlir-rohit-test 2022-11-16 01:15:06 +00:00
Ian Bearman
303790da88 [BUILD] use Python Var In Tests (#859) 2022-11-08 17:44:19 +00:00
Daniil Fukalov
ca4470fb96 [Triton-MLIR][ROCM] Temporary switch off some test lines.
Will get it back after a switch for ROCM addition.
2022-11-02 18:34:12 +03:00
Philippe Tillet
1e91ed30d0 [RUNTIME] Major code cleanup (#711)
This PR does the following:
- CUDA utilities (e.g., cuGetInfo) won't be compiled as part of libtriton.so anymore.
- Refactoring driver/llvm.cc to split it between PTX codegen and python.
- By extension this will also deprecate include/external so Triton won't have to live with a copy of some CUDA/Hip headers anymore.
- `triton-translate` becomes a `triton.tools.aot` Python utility that re-uses functions from the triton.compile sub-module.
2022-09-26 16:38:06 -07:00
goostavz
993ba7035a [BACKEND] Codegen bringup, index calculation of blocked_layout & support of LoadOp, BroadcastOp, ViewOp & MakeRangeOp (#38)
Co-authored-by: gzhu <gzhu@nvidia.com>
2022-08-14 19:58:59 -07:00
Yan Chunwei
83ef74f248 [BACKEND] Extracting numWarps from tritonGPU module (#39) 2022-08-08 09:40:20 -07:00
Yan Chunwei
920723cf3d [BACKEND] add triton-translate to translate mlir to llvmir or PTX code (#37) 2022-08-07 22:34:36 -07:00