| 267 | } |
| 268 | |
| 269 | bool RelayoutKernel::checkInputLayout(MemRefType memref, size_t index) { |
| 270 | int64_t offset; |
| 271 | llvm::SmallVector<int64_t> stride; |
| 272 | if (failed(getStridesAndOffset(memref, stride, offset))) { |
| 273 | return false; |
| 274 | } |
| 275 | |
| 276 | // could pass any strided memref to relayout |
| 277 | return true; |
| 278 | } |
| 279 | |
| 280 | #include "compiler/Dialect/Kernel/IR/KernelInterfaces.cpp.inc" |
| 281 |