[mlir] Expose simplifyAffineExpr through python api (#133926)

Author: Hardcode84

mlir/lib/Bindings/Python/IRAffine.cpp

@@ -599,6 +599,16 @@ void mlir::python::populateIRAffine(nb::module_ &m) {
           },
           nb::arg("num_symbols"), nb::arg("shift"),
           nb::arg("offset").none() = 0)
+      .def_static(
+          "simplify_affine_expr",
+          [](PyAffineExpr &self, uint32_t numDims, uint32_t numSymbols) {
+            return PyAffineExpr(
+                self.getContext(),
+                mlirSimplifyAffineExpr(self, numDims, numSymbols));
+          },
+          nb::arg("expr"), nb::arg("num_dims"), nb::arg("num_symbols"),
+          "Simplify an affine expression by flattening and some amount of "
+          "simple analysis.")
       .def_static(
           "get_add", &PyAffineAddExpr::get,
           "Gets an affine expression containing a sum of two expressions.")