mirror of https://github.com/llvm/torch-mlir
DecomposeComplexOps: update parseEquation to skip space char for AtenEinsumOp op (#2910)
Just a minor update to skip the space char if included in the equation string --------- Co-authored-by: Ze Zhang <ze.zhang@getcruise.com>pull/2839/merge
parent
77b7550997
commit
f3b38e5d12
|
@ -181,7 +181,7 @@ static bool parseEquation(const std::string &equation,
|
|||
inputToken.clear();
|
||||
currentVariable = kIsResult;
|
||||
index++;
|
||||
} else {
|
||||
} else if (equation[index] != ' ') {
|
||||
return false;
|
||||
}
|
||||
index++;
|
||||
|
|
Loading…
Reference in New Issue