diff options
Diffstat (limited to 'mlir/lib/Dialect/Traits.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Traits.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/lib/Dialect/Traits.cpp b/mlir/lib/Dialect/Traits.cpp index 6beb8cba41d..0fa57c23364 100644 --- a/mlir/lib/Dialect/Traits.cpp +++ b/mlir/lib/Dialect/Traits.cpp @@ -86,7 +86,7 @@ Type OpTrait::util::getBroadcastedType(Type type1, Type type2) { } // Returns the shape of the given type. - auto getShape = [](Type type) -> ArrayRef<int> { + auto getShape = [](Type type) -> ArrayRef<int64_t> { if (auto vtType = type.dyn_cast<VectorOrTensorType>()) return vtType.getShape(); return {}; @@ -104,7 +104,7 @@ Type OpTrait::util::getBroadcastedType(Type type1, Type type2) { // The result shape has the maximum among the two inputs at every // dimension index. - SmallVector<int, 4> resultShape; + SmallVector<int64_t, 4> resultShape; if (shape1.size() > shape2.size()) { std::copy(shape1.begin(), shape1.end(), std::back_inserter(resultShape)); } else { |
