llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[TorchToLinalg] Simplify QuantizePerTensor lowering (#3576) 

Uses arith::MaximumFOp and arith::MinimumFOp instead of comparison and
select ops to improve readability of IR.
pull/3608/head
zjgarvey 2024-08-02 11:40:52 -07:00 committed by GitHub
parent f7b5c13870
commit 79ae0afc2f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/Conversion/TorchToLinalg/Uncategorized.cpp
View File

@ -1461,12 +1461,8 @@ static Value createLinalgPayloadCalculationForElementwiseOp(
b.create<arith::ConstantOp>(loc, b.getFloatAttr(valueTy, minI)); b.create<arith::ConstantOp>(loc, b.getFloatAttr(valueTy, minI));
Value maxVal = Value maxVal =
b.create<arith::ConstantOp>(loc, b.getFloatAttr(valueTy, maxI)); b.create<arith::ConstantOp>(loc, b.getFloatAttr(valueTy, maxI));
Value minCmp = value = b.create<arith::MaximumFOp>(loc, value, minVal);
b.create<arith::CmpFOp>(loc, arith::CmpFPredicate::ULT, value, minVal); value = b.create<arith::MinimumFOp>(loc, value, maxVal);
Value maxCmp =
b.create<arith::CmpFOp>(loc, arith::CmpFPredicate::UGT, value, maxVal);
value = b.create<arith::SelectOp>(loc, minCmp, minVal, value);
value = b.create<arith::SelectOp>(loc, maxCmp, maxVal, value);
if (isUnsigned) { if (isUnsigned) {
value = b.create<arith::FPToUIOp>(loc, destTy, value); value = b.create<arith::FPToUIOp>(loc, destTy, value);