| 17 | namespace mlir { |
| 18 | namespace Kernel { |
| 19 | bool isContiguous(MemRefType memref) { |
| 20 | int64_t offset; |
| 21 | llvm::SmallVector<int64_t> stride; |
| 22 | if (failed(getStridesAndOffset(memref, stride, offset))) { |
| 23 | return false; |
| 24 | } |
| 25 | |
| 26 | auto shape = memref.getShape(); |
| 27 | |
| 28 | if (stride.size() != shape.size()) |
| 29 | return false; |
| 30 | |
| 31 | int64_t cur = 1; |
| 32 | for (int i = stride.size() - 1; i >= 0; --i) { |
| 33 | auto dim = shape[i]; |
| 34 | if (dim > 1 && cur != stride[i]) { |
| 35 | return false; |
| 36 | } |
| 37 | cur *= shape[i]; |
| 38 | } |
| 39 | return true; |
| 40 | } |
| 41 | } // namespace Kernel |
| 42 | } // namespace mlir |
| 43 | |