Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Signed-off-by: hanhanW <[email protected]>
  • Loading branch information
hanhanW committed Aug 5, 2024
1 parent 9cbf386 commit b5d41d5
Showing 1 changed file with 13 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -288,19 +288,19 @@ struct ConvertBf16ToUInt16BuffersPass final
vector::BroadcastOp, vector::ShuffleOp, vector::ExtractElementOp,
vector::ExtractOp, vector::InsertElementOp, vector::InsertOp,
vector::ScalableInsertOp, vector::ScalableExtractOp,
vector::InsertStridedSliceOp, vector::ReshapeOp,
vector::ExtractStridedSliceOp, vector::TransferReadOp,
vector::TransferWriteOp, vector::LoadOp, vector::StoreOp,
vector::MaskedLoadOp, vector::MaskedStoreOp, vector::GatherOp,
vector::ScatterOp, vector::ExpandLoadOp, vector::CompressStoreOp,
vector::ShapeCastOp, vector::ConstantMaskOp, vector::CreateMaskOp,
vector::MaskOp, vector::TransposeOp, vector::FlatTransposeOp,
vector::SplatOp, vector::YieldOp>([&typeConverter](Operation *op) {
bool legal = typeConverter.isLegal(op);
LLVM_DEBUG(if (!legal) llvm::dbgs()
<< "Bf16Emulation: illegal op: " << *op << "\n");
return legal;
});
vector::InsertStridedSliceOp, vector::ExtractStridedSliceOp,
vector::TransferReadOp, vector::TransferWriteOp, vector::LoadOp,
vector::StoreOp, vector::MaskedLoadOp, vector::MaskedStoreOp,
vector::GatherOp, vector::ScatterOp, vector::ExpandLoadOp,
vector::CompressStoreOp, vector::ShapeCastOp, vector::ConstantMaskOp,
vector::CreateMaskOp, vector::MaskOp, vector::TransposeOp,
vector::FlatTransposeOp, vector::SplatOp, vector::YieldOp>(
[&typeConverter](Operation *op) {
bool legal = typeConverter.isLegal(op);
LLVM_DEBUG(if (!legal) llvm::dbgs()
<< "Bf16Emulation: illegal op: " << *op << "\n");
return legal;
});

RewritePatternSet patterns(ctx);
populateIreeBf16EmulationPatterns(patterns, typeConverter);
Expand Down

0 comments on commit b5d41d5

Please sign in to comment.