Implemnet ValueBound interface for the op

Signed-off-by: hanhanW <hanhan0912@gmail.com>

Author: hanhanW

compiler/src/iree/compiler/Codegen/Dialect/VectorExt/IR/BUILD.bazel

@@ -38,6 +38,7 @@ iree_td_library(
         "@llvm-project//mlir:InferTypeOpInterfaceTdFiles",
         "@llvm-project//mlir:OpBaseTdFiles",
         "@llvm-project//mlir:SideEffectInterfacesTdFiles",
+        "@llvm-project//mlir:ValueBoundsOpInterfaceTdFiles",
         "@llvm-project//mlir:VectorInterfacesTdFiles",
     ],
 )
@@ -82,6 +83,7 @@ iree_compiler_cc_library(
         "@llvm-project//mlir:SideEffectInterfaces",
         "@llvm-project//mlir:Support",
         "@llvm-project//mlir:TensorDialect",
+        "@llvm-project//mlir:ValueBoundsOpInterface",
         "@llvm-project//mlir:VectorDialect",
         "@llvm-project//mlir:VectorInterfaces",
     ],

compiler/src/iree/compiler/Codegen/Dialect/VectorExt/IR/CMakeLists.txt

@@ -45,6 +45,7 @@ iree_cc_library(
     MLIRSideEffectInterfaces
     MLIRSupport
     MLIRTensorDialect
+    MLIRValueBoundsOpInterface
     MLIRVectorDialect
     MLIRVectorInterfaces
     iree::compiler::Codegen::Utils::VectorOpUtils

compiler/src/iree/compiler/Codegen/Dialect/VectorExt/IR/VectorExtDialect.h

@@ -14,6 +14,7 @@
 #include "mlir/IR/OpDefinition.h"
 #include "mlir/Interfaces/ControlFlowInterfaces.h"
 #include "mlir/Interfaces/SideEffectInterfaces.h"
+#include "mlir/Interfaces/ValueBoundsOpInterface.h"
 #include "mlir/Interfaces/VectorInterfaces.h"
 
 // clang-format off: must be included after all LLVM/MLIR headers

compiler/src/iree/compiler/Codegen/Dialect/VectorExt/IR/VectorExtOps.cpp

@@ -11,6 +11,7 @@
 #include "mlir/Dialect/Vector/IR/VectorOps.h"
 #include "mlir/IR/Matchers.h"
 #include "mlir/IR/TypeUtilities.h"
+#include "mlir/Interfaces/ValueBoundsOpInterface.h"
 
 using namespace mlir;
 using namespace mlir::iree_compiler::IREE::VectorExt;
@@ -26,6 +27,11 @@ LogicalResult ToLayoutOp::verify() {
   return getLayout().isValidLayout(getInput().getType(), getLoc());
 }
 
+void ToLayoutOp::populateBoundsForShapedValueDim(
+    Value value, int64_t dim, ValueBoundsConstraintSet &cstr) {
+  cstr.bound(value)[dim] == cstr.getExpr(getInput(), dim);
+}
+
 // to_simd -> to_simt
 OpFoldResult ToSIMDOp::fold(FoldAdaptor) {
   if (auto simtOp = getOperand().getDefiningOp<ToSIMTOp>()) {

compiler/src/iree/compiler/Codegen/Dialect/VectorExt/IR/VectorExtOps.td

@@ -14,6 +14,7 @@ include "iree/compiler/Codegen/Dialect/VectorExt/IR/VectorExtInterfaces.td"
 include "mlir/IR/OpBase.td"
 include "mlir/Interfaces/ControlFlowInterfaces.td"
 include "mlir/Interfaces/SideEffectInterfaces.td"
+include "mlir/Interfaces/ValueBoundsOpInterface.td"
 
 //===----------------------------------------------------------------------===//
 // Base class.
@@ -29,7 +30,9 @@ class IREEVectorExt_PureOp<string mnemonic, list<Trait> traits = []> :
 
 def IREEVectorExt_ToLayoutOp : IREEVectorExt_PureOp<"to_layout", [
   Pure,
-  AllTypesMatch<["input", "output"]>
+  AllTypesMatch<["input", "output"]>,
+  DeclareOpInterfaceMethods<ValueBoundsOpInterface,
+    ["populateBoundsForShapedValueDim"]>
   ]> {
   let summary = [{Layout conversion operator.}];
   let description = [{