stablehlo-legalize-to-linalg not exposed to Python bindings

[Wheest]

Request description

Using the latest version (165cd60cfa at time of writing), the stablehlo-legalize-to-linalg pass is not exposed in the Python bindings.

with open(input_mlir_file, "r") as f:
    mlir_text = f.read()

with Context() as ctx:
    stablehlo.register_dialect(ctx)
    stablehlo.register_stablehlo_passes()


    module = Module.parse(mlir_text)

    pm = PassManager.parse(
        "builtin.module(func.func("
        "shape-legalize-to-stablehlo,"
        "stablehlo-aggressive-folder,"
        "stablehlo-aggressive-simplification,"
        "stablehlo-legalize-to-linalg"
        "))"
    )

    pm.run(module.operation)

Fails with ValueError: MLIR Textual PassPipeline Parser:1:103: error: 'stablehlo-legalize-to-linalg' does not refer to a registered pass or pass pipeline, whereas removing this pass the rest behave as normal.

The pass is available under the stablehlo-opt binary.

[GleasonK]

This can probably be resolved by adding conversion pass registration to:

stablehlo/stablehlo/integrations/c/StablehloPasses.cpp

Line 17 in 7b13f60

void mlirRegisterAllStablehloPasses() { mlir::stablehlo::registerPasses(); }

Can try to get to this in the near future but feel free to submit a PR if you'd like it sooner