diff --git a/mlir/include/mlir/ExecutionEngine/CRunnerUtils.h b/mlir/include/mlir/ExecutionEngine/CRunnerUtils.h --- a/mlir/include/mlir/ExecutionEngine/CRunnerUtils.h +++ b/mlir/include/mlir/ExecutionEngine/CRunnerUtils.h @@ -167,6 +167,7 @@ //===----------------------------------------------------------------------===// // Small runtime support "lib" for vector.print lowering during codegen. //===----------------------------------------------------------------------===// +extern "C" MLIR_CRUNNERUTILS_EXPORT void print_i1(bool b); extern "C" MLIR_CRUNNERUTILS_EXPORT void print_i32(int32_t i); extern "C" MLIR_CRUNNERUTILS_EXPORT void print_i64(int64_t l); extern "C" MLIR_CRUNNERUTILS_EXPORT void print_f32(float f); diff --git a/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp b/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp --- a/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp +++ b/mlir/lib/Conversion/VectorToLLVM/ConvertVectorToLLVM.cpp @@ -925,7 +925,9 @@ Type eltType = vectorType ? vectorType.getElementType() : printType; int64_t rank = vectorType ? vectorType.getRank() : 0; Operation *printer; - if (eltType.isSignlessInteger(32)) + if (eltType.isSignlessInteger(1)) + printer = getPrintI1(op); + else if (eltType.isSignlessInteger(32)) printer = getPrintI32(op); else if (eltType.isSignlessInteger(64)) printer = getPrintI64(op); @@ -992,6 +994,11 @@ } // Helpers for method names. + Operation *getPrintI1(Operation *op) const { + LLVM::LLVMDialect *dialect = typeConverter.getDialect(); + return getPrint(op, dialect, "print_i1", + LLVM::LLVMType::getInt1Ty(dialect)); + } Operation *getPrintI32(Operation *op) const { LLVM::LLVMDialect *dialect = typeConverter.getDialect(); return getPrint(op, dialect, "print_i32", diff --git a/mlir/lib/ExecutionEngine/CRunnerUtils.cpp b/mlir/lib/ExecutionEngine/CRunnerUtils.cpp --- a/mlir/lib/ExecutionEngine/CRunnerUtils.cpp +++ b/mlir/lib/ExecutionEngine/CRunnerUtils.cpp @@ -23,6 +23,7 @@ // By providing elementary printing methods only, this // library can remain fully unaware of low-level implementation // details of our vectors. Also useful for direct LLVM IR output. +extern "C" void print_i1(bool b) { fputc(b ? '1' : '0', stdout); } extern "C" void print_i32(int32_t i) { fprintf(stdout, "%" PRId32, i); } extern "C" void print_i64(int64_t l) { fprintf(stdout, "%" PRId64, l); } extern "C" void print_f32(float f) { fprintf(stdout, "%g", f); } diff --git a/mlir/test/Conversion/VectorToLLVM/vector-to-llvm.mlir b/mlir/test/Conversion/VectorToLLVM/vector-to-llvm.mlir --- a/mlir/test/Conversion/VectorToLLVM/vector-to-llvm.mlir +++ b/mlir/test/Conversion/VectorToLLVM/vector-to-llvm.mlir @@ -418,6 +418,15 @@ // CHECK: llvm.mlir.constant(0 : index // CHECK: llvm.insertvalue {{.*}}[2] : !llvm<"{ [8 x [8 x <8 x float>]]*, [8 x [8 x <8 x float>]]*, i64 }"> +func @vector_print_scalar_i1(%arg0: i1) { + vector.print %arg0 : i1 + return +} +// CHECK-LABEL: llvm.func @vector_print_scalar_i1( +// CHECK-SAME: %[[A:.*]]: !llvm.i1) +// CHECK: llvm.call @print_i1(%[[A]]) : (!llvm.i1) -> () +// CHECK: llvm.call @print_newline() : () -> () + func @vector_print_scalar_i32(%arg0: i32) { vector.print %arg0 : i32 return