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

[NFC] Minor NFC changes 

Helping make code clean.
pull/2381/head
Abdul Raheem Beigh 2023-08-07 12:52:03 +01:00
parent 8d4f4cb104
commit b457b051db
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Dialect/Torch/Transforms/ReifyDtypeCalculations.cpp
View File

@ -44,10 +44,12 @@ dtypeFunctionArgsBuilder(OpBuilder &b, Location loc,
SmallVector<Value> dtypeFuncArgs; SmallVector<Value> dtypeFuncArgs;
ArrayRef<Type> desiredTypes = dtypeFunc.getArgumentTypes(); ArrayRef<Type> desiredTypes = dtypeFunc.getArgumentTypes();
for (auto operand : originalOperands) { for (Value operand : originalOperands) {
assert(!desiredTypes.empty() && assert(!desiredTypes.empty() &&
"`dtypeFunc` should have at least one argument for each argument in " "`dtypeFunc` should have at least one argument for each argument in "
"`originalOperands`"); "`originalOperands`");
// Getting hold of desired types and accordingly adjusting function
// arguments
Type desiredType = desiredTypes.front(); Type desiredType = desiredTypes.front();
FailureOr<Value> otherArg; FailureOr<Value> otherArg;
if (failed(otherArg = adjustFunctionArg(b, loc, operand, desiredType, if (failed(otherArg = adjustFunctionArg(b, loc, operand, desiredType,