Skip to content

Commit 9814aec

Browse files
committed
expand types
1 parent 246ba0f commit 9814aec

File tree

2 files changed

+15
-14
lines changed

2 files changed

+15
-14
lines changed

mlir/lib/Dialect/GPU/IR/GPUDialect.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2274,12 +2274,14 @@ KernelTableAttr::verify(function_ref<InFlightDiagnostic()> emitError,
22742274
}
22752275

22762276
KernelAttr KernelTableAttr::lookup(StringRef key) const {
2277-
auto it = impl::findAttrSorted(begin(), end(), key);
2277+
std::pair<ArrayRef<KernelAttr>::iterator, bool> it =
2278+
impl::findAttrSorted(begin(), end(), key);
22782279
return it.second ? *it.first : KernelAttr();
22792280
}
22802281

22812282
KernelAttr KernelTableAttr::lookup(StringAttr key) const {
2282-
auto it = impl::findAttrSorted(begin(), end(), key);
2283+
std::pair<ArrayRef<KernelAttr>::iterator, bool> it =
2284+
impl::findAttrSorted(begin(), end(), key);
22832285
return it.second ? *it.first : KernelAttr();
22842286
}
22852287

mlir/lib/Target/LLVM/ROCDL/Utils.cpp

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -119,18 +119,18 @@ static Attribute convertNode(Builder &builder,
119119

120120
static Attribute convertNode(Builder &builder,
121121
llvm::msgpack::ArrayDocNode &node) {
122-
using NodeKind = llvm::msgpack::Type;
123122
// Use `DenseIntAttr` if we know all the attrs are ints.
124123
if (llvm::all_of(node, [](llvm::msgpack::DocNode &n) {
125-
auto kind = n.getKind();
126-
return kind == NodeKind::Int || kind == NodeKind::UInt;
124+
llvm::msgpack::Type kind = n.getKind();
125+
return kind == llvm::msgpack::Type::Int ||
126+
kind == llvm::msgpack::Type::UInt;
127127
})) {
128128
SmallVector<int64_t> values;
129129
for (llvm::msgpack::DocNode &n : node) {
130130
llvm::msgpack::Type kind = n.getKind();
131-
if (kind == NodeKind::Int)
131+
if (kind == llvm::msgpack::Type::Int)
132132
values.push_back(n.getInt());
133-
else if (kind == NodeKind::UInt)
133+
else if (kind == llvm::msgpack::Type::UInt)
134134
values.push_back(n.getUInt());
135135
}
136136
return builder.getDenseI64ArrayAttr(values);
@@ -148,19 +148,18 @@ static Attribute convertNode(Builder &builder,
148148

149149
static Attribute convertNode(Builder &builder, llvm::msgpack::DocNode &node) {
150150
using namespace llvm::msgpack;
151-
using NodeKind = llvm::msgpack::Type;
152151
switch (node.getKind()) {
153-
case NodeKind::Int:
152+
case llvm::msgpack::Type::Int:
154153
return builder.getI64IntegerAttr(node.getInt());
155-
case NodeKind::UInt:
154+
case llvm::msgpack::Type::UInt:
156155
return builder.getI64IntegerAttr(node.getUInt());
157-
case NodeKind::Boolean:
156+
case llvm::msgpack::Type::Boolean:
158157
return builder.getI64IntegerAttr(node.getBool());
159-
case NodeKind::String:
158+
case llvm::msgpack::Type::String:
160159
return builder.getStringAttr(node.getString());
161-
case NodeKind::Array:
160+
case llvm::msgpack::Type::Array:
162161
return convertNode(builder, node.getArray());
163-
case NodeKind::Map:
162+
case llvm::msgpack::Type::Map:
164163
return convertNode(builder, node.getMap());
165164
default:
166165
return nullptr;

0 commit comments

Comments
 (0)