Commit 40e8ad40 authored by Mccaskey, Alex's avatar Mccaskey, Alex
Browse files

adding memref rank check on new printoplowering code 



Signed-off-by: Mccaskey, Alex's avatarAlex McCaskey <mccaskeyaj@ornl.gov>
parent efbe54e8

mlir/transforms/lowering/PrintOpLowering.cpp

+2 −0
Original line number Diff line number Diff line
@@ -106,6 +106,7 @@ LogicalResult PrintOpLowering::matchAndRewrite( 
    } else if (auto mem_ref_type =

                   operand.getType().dyn_cast_or_null<mlir::MemRefType>()) {

      if (mem_ref_type.getElementType().isa<mlir::IntegerType>() &&

          mem_ref_type.getRank() > 0 &&

          mem_ref_type.getElementType().getIntOrFloatBitWidth() == 1) {

        // This is a bit array...

        auto dim = mem_ref_type.getShape()[0];
@@ -151,6 +152,7 @@ LogicalResult PrintOpLowering::matchAndRewrite( 
    } else if (auto mem_ref_type =

                   o.getType().dyn_cast_or_null<mlir::MemRefType>()) {

      if (mem_ref_type.getElementType().isa<mlir::IntegerType>() &&

          mem_ref_type.getRank() > 0 &&

          mem_ref_type.getElementType().getIntOrFloatBitWidth() == 1) {

        // This is a bit array...

        auto dim = mem_ref_type.getShape()[0];