[mlir] Expose Arith::ExpandOps options in constructor

[rsuderman]

Summary

ExpandOpsPass could only be configured via command line flags. Updated
to allowed constructing using the specified Options structure.

[zero9178]

FYI, this code could be automatically generated by TableGen (https://github.com/llvm/llvm-project/issues/57475) by simply removing the let constructor = ...; field assignment in the Passes.td file

mlir/include/mlir/Dialect/Arith/Transforms/Passes.h

 void populateExpandBFloat16Patterns(RewritePatternSet &patterns);
 
 /// Add patterns to expand Arith ops.
 void populateArithExpandOpsPatterns(RewritePatternSet &patterns);
 
 /// Create a pass to legalize Arith ops.
 std::unique_ptr<Pass> createArithExpandOpsPass();
 
+/// Create a pass to legalize Arith ops with specified configuration.
+std::unique_ptr<Pass>
+createArithExpandOpsPass(const ArithExpandOpsOptions &options);
+
 /// Create a pass to replace signed ops with unsigned ones where they are proven
 /// equivalent.
 std::unique_ptr<Pass> createArithUnsignedWhenEquivalentPass();
 
 /// Add patterns for int range based optimizations.
 void populateIntRangeOptimizationsPatterns(RewritePatternSet &patterns,
                                            DataFlowSolver &solver);
 

mlir/lib/Dialect/Arith/Transforms/ExpandOps.cpp

LogicalResult matchAndRewrite(arith::TruncFOp op,
 
     rewriter.replaceOp(op, result);
     return success();
   }
 };
 
 struct ArithExpandOpsPass
     : public arith::impl::ArithExpandOpsBase<ArithExpandOpsPass> {
+  ArithExpandOpsPass() = default;
+  ArithExpandOpsPass(const arith::ArithExpandOpsOptions& options) {
+    this->includeBf16 = options.includeBf16;
+  }
+
   void runOnOperation() override {
     RewritePatternSet patterns(&getContext());
     ConversionTarget target(getContext());
 
     arith::populateArithExpandOpsPatterns(patterns);
 
     target.addLegalDialect<arith::ArithDialect>();
     // clang-format off
patterns.add<
     MaxMinFOpConverter<MinFOp, arith::CmpFPredicate::ULT>
    >(patterns.getContext());
   // clang-format on
 }
 
 std::unique_ptr<Pass> mlir::arith::createArithExpandOpsPass() {
   return std::make_unique<ArithExpandOpsPass>();
 }
+
+std::unique_ptr<Pass> mlir::arith::createArithExpandOpsPass(
+  const ArithExpandOpsOptions& options) {
+  return std::make_unique<ArithExpandOpsPass>(options);
+}