Register QuantizationDialect (#1978)

Bug fix to register `mlir::quant::QuantizationDialect`.
openxla · Feb 1, 2024 · aeeca3c · aeeca3c
1 parent 39c4a00
commit aeeca3c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/stablehlo/dialect/Register.cpp b/stablehlo/dialect/Register.cpp
@@ -31,6 +31,7 @@ namespace stablehlo {
 void registerAllDialects(mlir::DialectRegistry &registry) {
   // clang-format off
   registry.insert<mlir::func::FuncDialect,
+                  mlir::quant::QuantizationDialect,
                   mlir::sparse_tensor::SparseTensorDialect>();
   registry.insert<mlir::chlo::ChloDialect,
                   mlir::stablehlo::StablehloDialect,