Yan Chunwei
|
920723cf3d
|
[BACKEND] add triton-translate to translate mlir to llvmir or PTX code (#37)
|
2022-08-07 22:34:36 -07:00 |
|
Philippe Tillet
|
78ebbe24c7
|
[FRONTEND] Added ExpandDimsOp primitive (#36)
|
2022-08-04 18:41:06 -07:00 |
|
Keren Zhou
|
a7b49b3227
|
[BACKEND] Memory allocation (#33)
|
2022-08-04 11:22:49 -07:00 |
|
Yan Chunwei
|
b988bae813
|
Init TritonGPU to LLVM dialect conversion (#32)
* add toLLVM pass
* update num-warps setting in mlir
|
2022-08-04 10:15:45 +08:00 |
|
Philippe Tillet
|
3236642e8f
|
[OPTIMIZER] Added memory coalescing pass (#31)
|
2022-07-31 20:59:31 -07:00 |
|
Philippe Tillet
|
d1593e6ca8
|
[TritonGPU] Improved documentation and semantics of layout encodings (#30)
|
2022-07-31 13:59:44 -07:00 |
|
Philippe Tillet
|
6d62d88d4f
|
[CI] run clang-format (#24)
|
2022-07-26 17:25:03 -07:00 |
|
Philippe Tillet
|
3265e0df5a
|
[PYTHON] Cleaned up legacy code; added simple standalone compilation API (#22)
|
2022-07-26 11:06:45 -07:00 |
|
Philippe Tillet
|
27c9f3d8cb
|
[FRONTEND] Added comment on TensorSizeTrait::maxElement (#20)
|
2022-07-25 01:18:45 -07:00 |
|
Philippe Tillet
|
a633d2b403
|
[Analysis] Added Axis Info Analysis (#8)
|
2022-07-19 13:38:48 -07:00 |
|
Yan Da
|
9d1b5e3f79
|
special encoding for broadcast
|
2022-06-18 21:16:45 +08:00 |
|
Yan Da
|
53cf93ce6a
|
Revert "Remove TypeConverter from TritonToTritonGPU conversion"
This reverts commit 64d0b87ef0.
|
2022-06-18 14:57:41 +08:00 |
|
Yan Da
|
64d0b87ef0
|
Remove TypeConverter from TritonToTritonGPU conversion
|
2022-06-18 14:34:59 +08:00 |
|
Yan Da
|
560e29229b
|
register conversion in triton-opt
|
2022-06-07 19:33:51 +08:00 |
|
Yan Da
|
366dddc3bc
|
update mma encoding & triton-opt
|
2022-06-06 21:03:58 +08:00 |
|
Yan Da
|
7807f64ef3
|
rename sharded_layout => blocked_layout
|
2022-06-05 16:14:59 +08:00 |
|
Da Yan
|
e36a54eb86
|
more progress on the definition of layouts
|
2022-05-31 11:43:21 +00:00 |
|
Yan Da
|
441fd7c3cc
|
assembly format
|
2022-05-25 17:53:24 +08:00 |
|
Yan Da
|
9b670cfb9f
|
Add ReduceOp
|
2022-05-25 14:15:36 +08:00 |
|
Yan Da
|
a2c9f919a8
|
TritonGPU verifier
|
2022-05-24 19:48:56 +08:00 |
|
Yan Da
|
36c45ec687
|
make numStages an option in PipelinePass
|
2022-05-23 12:47:55 +08:00 |
|
Yan Da
|
c3c4ac3733
|
TritonGPU combiner
|
2022-05-16 19:17:15 +08:00 |
|
Yan Da
|
e3916c3a46
|
TritonGPU combiner
|
2022-05-16 19:16:01 +08:00 |
|
Yan Da
|
7027af9666
|
The pipeline pass is now functional
|
2022-05-15 22:29:27 +08:00 |
|
Yan Da
|
d23d7b244c
|
More on the pipeline pass
|
2022-05-11 20:31:08 +08:00 |
|
Yan Da
|
1a4fbed25b
|
Skeleton for the pipeline pass
|
2022-05-11 16:13:53 +08:00 |
|
Yan Da
|
2d281cbc0a
|
ConstantOp conversion pattern
|
2022-05-04 15:35:43 +08:00 |
|
Yan Da
|
b9279d2e3b
|
More progress on TritonGPU conversion
|
2022-05-04 14:54:31 +08:00 |
|
Yan Da
|
75d32e2442
|
More on TritonGPU conversion
|
2022-05-02 21:51:00 +08:00 |
|
Yan Da
|
1428185c9c
|
More progress on TritonGPUTypeConverter & TritonGPUConversionTarget
|
2022-05-01 22:06:54 +08:00 |
|
Yan Da
|
4ece9fd1f3
|
Move dependentDialects from .cpp to .td
|
2022-05-01 13:06:51 +08:00 |
|
Phil Tillet
|
2c6a213131
|
[TRITONGPU] Added template for Triton -> TritonGPU conversion
|
2022-04-30 16:00:39 -07:00 |
|
Yan Da
|
2239ac1998
|
more progress on TritonGPU
|
2022-04-28 18:51:31 +08:00 |
|
Philippe Tillet
|
012e8c5b2b
|
fixup
|
2022-04-27 16:39:27 -07:00 |
|
Philippe Tillet
|
513bcaee50
|
Added some ASCII art for encoding documentation
|
2022-04-27 16:28:27 -07:00 |
|
Yan Da
|
38d13ae618
|
Some progress on TritonGPU
|
2022-04-27 21:16:45 +08:00 |
|
Yan Da
|
edca91bf8f
|
Update traits (NoSideEffect)
|
2022-04-27 19:41:07 +08:00 |
|
Yan Da
|
8dfe78f6cf
|
Add TritonCombineOps
|
2022-04-27 19:28:21 +08:00 |
|
Yan Da
|
c70f6b666e
|
Merge previous changes
|
2022-04-27 14:06:55 +08:00 |
|
Yan Da
|
74585fb970
|
Add Triton CombineOps
|
2022-04-27 13:45:56 +08:00 |
|
Philippe Tillet
|
81001d318c
|
Putting Triton dialect in its own folder
|
2022-04-26 14:39:27 -07:00 |
|
Philippe Tillet
|
62a64ff29b
|
Fixed Python link bug in CMakeLists
|
2022-04-26 14:39:18 -07:00 |
|
Yan Da
|
fcbbb3c10e
|
Fix visit_While issues
|
2022-04-10 16:16:13 +08:00 |
|
Yan Da
|
f1cc67bbc3
|
triton -> tt
|
2022-04-10 12:07:19 +08:00 |
|
Yan Da
|
28e96bbfd1
|
Remove the dependency on TensorDialect
|
2022-04-08 19:43:09 +08:00 |
|
Yan Da
|
62f7609612
|
More on type inference & assembly format
|
2022-04-08 19:37:57 +08:00 |
|
Yan Da
|
13aead4808
|
Use TableGen to define new types
|
2022-04-08 16:32:46 +08:00 |
|
Yan Da
|
6002340456
|
Better textual representation
|
2022-04-07 20:44:41 +08:00 |
|
Yan Da
|
62f772123c
|
now kernel functions return nothing (instead of none)
|
2022-04-07 20:22:17 +08:00 |
|
Yan Da
|
040a2b6c75
|
Fix OpBuilder
|
2022-04-07 20:01:31 +08:00 |
|