Skip to content

Commit 4b3d519

Browse files
authored
[infrt] fix infrt script bug and function error. test=develop (#38384)
1 parent fc0a50a commit 4b3d519

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

paddle/infrt/dialect/infrt_base.h

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,13 @@ static mlir::IntegerAttr createI32Attr(mlir::OpBuilder &b, // NOLINT
5858
return b.getIntegerAttr(b.getI32Type(), constant);
5959
}
6060

61-
static mlir::ValueRange cvtValueToValueRange(const mlir::Value &operand) {
62-
return mlir::ValueRange(operand);
61+
static mlir::SmallVector<::mlir::Value, 4> cvtValueToValueRange(
62+
const mlir::Value &operand) {
63+
return mlir::SmallVector<::mlir::Value, 4>(1, operand);
6364
}
6465

65-
static mlir::ValueRange concatTwoValueRange(mlir::ValueRange operand_0,
66-
mlir::ValueRange operand_1) {
66+
static mlir::SmallVector<::mlir::Value, 4> concatTwoValueRange(
67+
mlir::ValueRange operand_0, mlir::ValueRange operand_1) {
6768
mlir::SmallVector<::mlir::Value, 4> operands;
6869
operands.append(operand_0.begin(), operand_0.end());
6970
operands.append(operand_1.begin(), operand_1.end());

paddle/scripts/infrt_build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,4 +132,4 @@ function main() {
132132

133133
main $@
134134

135-
rm -rf tmp_dir
135+
rm -rf $tmp_dir

0 commit comments

Comments
 (0)