diff options
Diffstat (limited to 'mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h')
| -rw-r--r-- | mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h b/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h index 9077d2e..ce7d5d0 100644 --- a/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h +++ b/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h @@ -30,6 +30,9 @@ namespace mlir { namespace mesh { +using MeshAxis = int16_t; +using MeshAxesAttr = DenseI16ArrayAttr; + bool isReductionLoop(IteratorType iType); bool areReductionAndPartialMatch(IteratorType iType, Partial partial); |