diff --git a/mlir/include/mlir/Dialect/Tosa/IR/TosaOpBase.td b/mlir/include/mlir/Dialect/Tosa/IR/TosaOpBase.td --- a/mlir/include/mlir/Dialect/Tosa/IR/TosaOpBase.td +++ b/mlir/include/mlir/Dialect/Tosa/IR/TosaOpBase.td @@ -40,7 +40,7 @@ there will be tools to lower from the ML frameworks into TOSA. }]; - let dependentDialects = ["tensor::TensorDialect"]; + let dependentDialects = ["tensor::TensorDialect", "quant::QuantizationDialect"]; let cppNamespace = "mlir::tosa"; let hasConstantMaterializer = 1;