gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

CatOp

Browse Source
This commit is contained in:
Yan Da
2022-03-25 14:17:17 +08:00
parent d5612333c0
commit 5e117966d0
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
include/triton/ir/TritonOps.td
View File

@@ -145,7 +145,6 @@ def TT_GEPOp : TT_Op<"getelementptr", [NoSideEffect, SameOperandsAndResultShape]
//
// Shape Manipulation Ops
//
// def TT_CatOp : TT_Op<"cat", []>;
def TT_BroadcastOp : TT_Op<"broadcast", [SameOperandsAndResultElementType]> {
let summary = "broadcast";
@@ -154,6 +153,14 @@ def TT_BroadcastOp : TT_Op<"broadcast", [SameOperandsAndResultElementType]> {
let results = (outs TT_Type:$result);
}
def TT_CatOp : TT_Op<"cat", [SameOperandsAndResultElementType]> {
let summary = "concatenate 2 tensors";
let arguments = (ins TT_Tensor:$lhs, TT_Tensor:$rhs);
let results = (outs TT_Tensor:$result);
}
//
// builtin Ops
//