correct the def name string

openxla · Feb 22, 2024 · f37360b · f37360b
1 parent 3ba1d74
commit f37360b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stablehlo/dialect/Base.td b/stablehlo/dialect/Base.td
@@ -73,7 +73,7 @@ class StableHLO_UniformQuantizedUnsignedInt<int width>
            CPred<"!$_self.cast<mlir::quant::UniformQuantizedType>()" #
                  ".isSigned()">]>,
     "QUI" # width # " type"> {
-  string name = "UniformQuantizedPerAxisUnsignedInt";
+  string name = "UniformQuantizedUnsignedInt";
   int bitwidth = width;
 }
 
@@ -84,7 +84,7 @@ class StableHLO_UniformQuantizedPerAxisUnsignedInt<int width>
            CPred<"!$_self.cast<mlir::quant::UniformQuantizedPerAxisType>()" #
                  ".isSigned()">]>,
     "QUI" # width # " type"> {
-  string name = "UniformQuantizedUnsignedInt";
+  string name = "UniformQuantizedPerAxisUnsignedInt";
   int bitwidth = width;
 }