This is an archive of the discontinued LLVM Phabricator instance.

Differential D140595

[mlir][sparse] add missing dependent dialect.
ClosedPublic

Authored by Peiming on Dec 22 2022, 5:37 PM.

Details

Reviewers
aartbik
nicolasvasilache
Commits
rG006340baf753: [mlir][sparse] add missing dependent dialect.

Diff Detail

Event Timeline

Peiming created this revision.Dec 22 2022, 5:37 PM
Herald added a reviewer: aartbik. · View Herald TranscriptDec 22 2022, 5:37 PM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: hanchung, jsetoain, Moerafaat and 22 others. · View Herald Transcript
Peiming requested review of this revision.Dec 22 2022, 5:37 PM
Herald added a reviewer: nicolasvasilache. · View Herald TranscriptDec 22 2022, 5:37 PM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
aartbik accepted this revision.Dec 22 2022, 5:38 PM
This revision is now accepted and ready to land.Dec 22 2022, 5:38 PM
This revision was landed with ongoing or failed builds.Dec 22 2022, 5:44 PM
Closed by commit rG006340baf753: [mlir][sparse] add missing dependent dialect. (authored by Peiming). · Explain Why
This revision was automatically updated to reflect the committed changes.
Peiming added a commit: rG006340baf753: [mlir][sparse] add missing dependent dialect..
Harbormaster completed remote builds in B204697: Diff 484997.Dec 22 2022, 5:54 PM

Revision Contents

PathSize
mlir/
lib/
Dialect/
SparseTensor/
Transforms/
2 lines

Diff 485001

mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp

Show All 10 Lines
#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h" #include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
#include "mlir/Dialect/Bufferization/IR/Bufferization.h" #include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/Bufferization/Transforms/Bufferize.h" #include "mlir/Dialect/Bufferization/Transforms/Bufferize.h"
#include "mlir/Dialect/Bufferization/Transforms/OneShotAnalysis.h" #include "mlir/Dialect/Bufferization/Transforms/OneShotAnalysis.h"
#include "mlir/Dialect/Bufferization/Transforms/OneShotModuleBufferize.h" #include "mlir/Dialect/Bufferization/Transforms/OneShotModuleBufferize.h"
#include "mlir/Dialect/Bufferization/Transforms/Passes.h" #include "mlir/Dialect/Bufferization/Transforms/Passes.h"
#include "mlir/Dialect/Bufferization/Transforms/Transforms.h" #include "mlir/Dialect/Bufferization/Transforms/Transforms.h"
#include "mlir/Dialect/Func/IR/FuncOps.h" #include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h" #include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/Dialect/SparseTensor/Transforms/Passes.h" #include "mlir/Dialect/SparseTensor/Transforms/Passes.h"
#include "mlir/Pass/PassManager.h" #include "mlir/Pass/PassManager.h"
#include "mlir/Transforms/Passes.h" #include "mlir/Transforms/Passes.h"
using namespace mlir; using namespace mlir;
using namespace mlir::func; using namespace mlir::func;
▲ Show 20 Lines • Show All 64 Lines • ▼ Show 20 Lines if (failed(bufferization::bufferizeOp(getOperation(), bufferizationOptions,
return failure(); return failure();
bufferization::removeBufferizationAttributesInModule(getOperation()); bufferization::removeBufferizationAttributesInModule(getOperation());
return success(); return success();
} }
void getDependentDialects(::mlir::DialectRegistry &registry) const override { void getDependentDialects(::mlir::DialectRegistry &registry) const override {
registry.insert<bufferization::BufferizationDialect>(); registry.insert<bufferization::BufferizationDialect>();
registry.insert<LLVM::LLVMDialect>();
} }
void runOnOperation() override { void runOnOperation() override {
{ {
// Run enabling transformations. // Run enabling transformations.
OpPassManager pm("builtin.module"); OpPassManager pm("builtin.module");
pm.addPass(createPreSparsificationRewritePass()); pm.addPass(createPreSparsificationRewritePass());
pm.addNestedPass<func::FuncOp>( pm.addNestedPass<func::FuncOp>(
▲ Show 20 Lines • Show All 82 Lines • Show Last 20 Lines