[mlir][tensor][bufferize] Load dependent dialects

[springerm]

Summary

Load dialects that will be generated by the extension. (Except for BufferizationDialect and MemrefDialect which are loaded already.)

mlir/lib/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.cpp

registry.addExtension(+[](MLIRContext *ctx, tensor::TensorDialect *dialect) {
     FromElementsOp::attachInterface<FromElementsOpInterface>(*ctx);
     GenerateOp::attachInterface<GenerateOpInterface>(*ctx);
     InsertOp::attachInterface<InsertOpInterface>(*ctx);
     InsertSliceOp::attachInterface<InsertSliceOpInterface>(*ctx);
     ParallelInsertSliceOp::attachInterface<ParallelInsertSliceOpInterface>(
         *ctx);
     RankOp::attachInterface<RankOpInterface>(*ctx);
     ReshapeOp::attachInterface<ReshapeOpInterface>(*ctx);
+
+    // Load additional dialects of which ops may get created.
+    ctx->loadDialect<arith::ArithmeticDialect, scf::SCFDialect>();
   });
 }