diff --git a/mlir/include/mlir/EDSC/Intrinsics.h b/mlir/include/mlir/EDSC/Intrinsics.h --- a/mlir/include/mlir/EDSC/Intrinsics.h +++ b/mlir/include/mlir/EDSC/Intrinsics.h @@ -213,6 +213,8 @@ using subi = ValueBuilder; using tanh = ValueBuilder; using view = ValueBuilder; +using zero_extendi = ValueBuilder; +using sign_extendi = ValueBuilder; /// Branches into the mlir::Block* captured by BlockHandle `b` with `operands`. /// diff --git a/mlir/test/EDSC/builder-api-test.cpp b/mlir/test/EDSC/builder-api-test.cpp --- a/mlir/test/EDSC/builder-api-test.cpp +++ b/mlir/test/EDSC/builder-api-test.cpp @@ -459,6 +459,32 @@ f.erase(); } +TEST_FUNC(zero_and_sign_extendi_op_i1_to_i8) { + using namespace edsc; + using namespace edsc::intrinsics; + using namespace edsc::op; + auto i1Type = IntegerType::get(1, &globalContext()); + auto i8Type = IntegerType::get(8, &globalContext()); + auto memrefType = MemRefType::get({}, i1Type, {}, 0); + auto f = makeFunction("zero_and_sign_extendi_op", {}, {memrefType, memrefType}); + + OpBuilder builder(f.getBody()); + ScopedContext scope(builder, f.getLoc()); + IndexedValue A(f.getArgument(0)); + IndexedValue B(f.getArgument(1)); + // clang-format off + edsc::intrinsics::zero_extendi(*A, i8Type); + edsc::intrinsics::sign_extendi(*B, i8Type); + // CHECK-LABEL: @zero_and_sign_extendi_op + // CHECK: %[[SRC1:.*]] = affine.load + // CHECK: zexti %[[SRC1]] : i1 to i8 + // CHECK: %[[SRC2:.*]] = affine.load + // CHECK: sexti %[[SRC2]] : i1 to i8 + // clang-format on + f.print(llvm::outs()); + f.erase(); +} + TEST_FUNC(select_op_i32) { using namespace edsc; using namespace edsc::intrinsics;