| commit | 5f5f71e73768cb185beaccdca463bf772cf8d1b3 | [log] [tgz] |
|---|---|---|
| author | Matthias Springer <springerm@google.com> | Mon Jul 25 11:30:22 2022 +0200 |
| committer | Matthias Springer <springerm@google.com> | Mon Jul 25 11:36:10 2022 +0200 |
| tree | 3f83455a326d2b8329c4d90c459a24e1a46ff1ca | |
| parent | d8800ead629d403a298d32e3b1a7ac4f7aac554a [diff] |
[mlir][tensor][bufferize] Load dependent dialects Load dialects that will be generated by the extension. (Except for BufferizationDialect and MemrefDialect which are loaded already.) Differential Revision: https://reviews.llvm.org/D130463
diff --git a/mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp b/mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp index 2916f1f..51bd6eb 100644 --- a/mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp +++ b/mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp
@@ -1063,5 +1063,8 @@ *ctx); RankOp::attachInterface<RankOpInterface>(*ctx); ReshapeOp::attachInterface<ReshapeOpInterface>(*ctx); + + // Load additional dialects of which ops may get created. + ctx->loadDialect<arith::ArithmeticDialect, scf::SCFDialect>(); }); }