Update 2 instances of isa<BlockArgument>

PiperOrigin-RevId: 252739405
This commit is contained in:
Nicolas Vasilache 2019-06-11 18:16:17 -07:00 committed by Mehdi Amini
parent 705b2b5ea4
commit d43b8923e0
1 changed files with 2 additions and 2 deletions

View File

@ -33,12 +33,12 @@ using namespace mlir::linalg;
using llvm::dbgs;
Value *Aliases::find(Value *v) {
if (v->getKind() == Value::Kind::BlockArgument)
if (isa<BlockArgument>(v))
return v;
auto it = aliases.find(v);
if (it != aliases.end()) {
assert(((it->getSecond()->getKind() == Value::Kind::BlockArgument &&
assert(((isa<BlockArgument>(it->getSecond()) &&
it->getSecond()->getType().isa<ViewType>()) ||
it->getSecond()->getType().isa<BufferType>()) &&
"Buffer or block argument expected");