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