From aeeca3c4f827131d6e2fd53c732618e54d515308 Mon Sep 17 00:00:00 2001
From: cryptodeal <39316528+cryptodeal@users.noreply.github.com>
Date: Thu, 1 Feb 2024 16:32:34 -0600
Subject: [PATCH] Register `QuantizationDialect` (#1978)

Bug fix to register `mlir::quant::QuantizationDialect`.
---
 stablehlo/dialect/Register.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/stablehlo/dialect/Register.cpp b/stablehlo/dialect/Register.cpp
index 79c67d43e06..0e87548f689 100644
--- 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,