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

[Triton-MLIR] [Frontend] Return a scalar if all input args are scalar (#839)

Browse Source
This commit is contained in:
Keren Zhou
2022-11-03 20:27:47 -07:00
committed by GitHub
parent 61f2ff98df
commit 4218e68d74
3 changed files with 60 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/triton/Dialect/Triton/IR/TritonOps.td
View File

@@ -357,12 +357,11 @@ def TT_ExtElemwiseOp : TT_Op<"ext_elemwise", [NoSideEffect, Elementwise, SameOpe
return $libpath/$libname:$symbol($args...)
}];
let arguments = (ins Variadic<TT_Tensor>:$args, StrAttr:$libname, StrAttr:$libpath, StrAttr:$symbol);
let arguments = (ins Variadic<TT_Type>:$args, StrAttr:$libname, StrAttr:$libpath, StrAttr:$symbol);
let results = (outs TT_Tensor:$result);
let results = (outs TT_Type:$result);
let assemblyFormat = "operands attr-dict `:` type(operands) `->` type($result)";
}
//