Skip to content

Commit

Permalink
[kernelgen] Don't try bufferizing TensorDialect
Browse files Browse the repository at this point in the history
This currently doesn't do anything because we
don't load the BufferizableOpInterface. But if
someone does we have a miscompile on our hands.

PiperOrigin-RevId: 617113554
  • Loading branch information
d0k authored and TensorFlow MLIR Team committed Mar 19, 2024
1 parent 5d36d84 commit e7b5c5f
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions transforms/bufferize_pass.cc
Original file line number Diff line number Diff line change
Expand Up @@ -153,8 +153,7 @@ struct ComputeOpAndFuncBufferizePass
// will be migrated to BufferizableOpInterface-based bufferization.
options.opFilter.allowDialect<bufferization::BufferizationDialect,
linalg::LinalgDialect, mhlo::MhloDialect,
shape::ShapeDialect, tensor::TensorDialect,
vector::VectorDialect>();
shape::ShapeDialect, vector::VectorDialect>();

if (failed(bufferization::bufferizeOp(getOperation(), options))) {
signalPassFailure();
Expand Down

0 comments on commit e7b5c5f

Please sign in to comment.