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

[BUILD] fix minor issues with MLIR assert enabled (#46)

Browse Source
This commit is contained in:
Shintaro Iwasaki
2022-08-11 21:20:47 -07:00
committed by GitHub
parent 3a48ca0d4d
commit 2ba9a83465
5 changed files with 53 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/Dialect/Triton/IR/Ops.cpp
View File

@@ -112,8 +112,19 @@ OpFoldResult BroadcastOp::fold(ArrayRef<Attribute> operands) {
auto constOperand = src().getDefiningOp<arith::ConstantOp>();
if (!constOperand)
return {};
auto shapedType = getType().cast<ShapedType>();
return SplatElementsAttr::get(shapedType, {constOperand.getValue()});
auto value = constOperand.getValue();
if (auto denseElemsAttr = value.dyn_cast<DenseElementsAttr>()) {
if (!denseElemsAttr.isSplat())
return {};
return SplatElementsAttr::get(shapedType,
denseElemsAttr.getSplatValue<Attribute>());
} else if (value.getType().isIntOrIndexOrFloat()) {
return SplatElementsAttr::get(shapedType, value);
} else {
return {};
}
}
} // namespace triton