From 5e117966d0ca70c92adb5498273782dceb2b8e40 Mon Sep 17 00:00:00 2001 From: Yan Da Date: Fri, 25 Mar 2022 14:17:17 +0800 Subject: [PATCH] CatOp --- include/triton/ir/TritonOps.td | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/include/triton/ir/TritonOps.td b/include/triton/ir/TritonOps.td index aaf44d949..4a6b8d5f1 100644 --- a/include/triton/ir/TritonOps.td +++ b/include/triton/ir/TritonOps.td @@ -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 //