weight_component = component;
        break;
      default:
        break;
    }
  }
  pass_manager.addNestedPass<mlir::func::FuncOp>(
      mlir::quant::stablehlo::CreateQuantizeWeightPass(weight_component));
}

}  // namespace quantization

namespace mlir::quant::stablehlo {

// Creates a pass that quantizes weight component of StableHLO graph.
std::unique_ptr<OperationPass<func::FuncOp>> CreateQuantizeWeightPass(
    const ::stablehlo::quantization::QuantizationComponentSpec&
        quantization_component_spec = {});

// Converts a serialized StableHLO module to bfloat16 and output serialized
// module.