Skip to content

Commit 9f7d82d

Browse files
Ayush AnshulAyush Anshul
authored andcommitted
call relevant getter for signed and unsigned ints
1 parent 07d5ce8 commit 9f7d82d

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/Dialect/ONNX/ONNXOps/OpHelper.cpp

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -599,7 +599,13 @@ RESULT_TYPE getScalarValue(ElementsAttr denseAttr, Type type) {
599599
if (elementaryType.isInteger(8) || elementaryType.isInteger(16) ||
600600
elementaryType.isInteger(32) || elementaryType.isInteger(64)) {
601601
auto valueIt = denseAttr.getValues<IntegerAttr>().begin();
602-
return static_cast<RESULT_TYPE>(mlir::cast<IntegerAttr>(*valueIt).getInt());
602+
if (type.isSignedInteger()) {
603+
return static_cast<RESULT_TYPE>(
604+
mlir::cast<IntegerAttr>(*valueIt).getSInt());
605+
} else {
606+
return static_cast<RESULT_TYPE>(
607+
mlir::cast<IntegerAttr>(*valueIt).getUInt());
608+
}
603609
} else if (mlir::isa<FloatType>(elementaryType)) {
604610
auto valueIt = denseAttr.getValues<APFloat>().begin();
605611
return static_cast<RESULT_TYPE>((*valueIt).convertToDouble());

0 commit comments

Comments
 (0)