Skip to content

Commit

Permalink
[StableHLO][NFC] Enable pow e2e test on llvm-cpu (iree-org#13792)
Browse files Browse the repository at this point in the history
This test is disabled on rv32 due to known issues with the musl lib. I
checked the generated code and it looks identical to the mhlo output
now.

Issue: iree-org#12678
  • Loading branch information
kuhar authored and nhasabni committed Aug 24, 2023
1 parent 558fd31 commit 97e6823
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 3 deletions.
1 change: 1 addition & 0 deletions build_tools/cmake/test_riscv.sh
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ if [[ "${RISCV_PLATFORM}-${RISCV_ARCH}" == "linux-riscv_32" ]]; then
# mhlo.power is also disabled because musl math library is not compiled for
# 32-bit.
test_exclude_args+=(
"stablehlo.*llvm-cpu.*pow"
"xla.*llvm-cpu.*pow"
)
fi
Expand Down
5 changes: 2 additions & 3 deletions tests/e2e/stablehlo_ops/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ iree_check_single_backend_test_suite(
"multiply.mlir",
"negate.mlir",
"pad.mlir",
"pow.mlir",
"reduce.mlir",
"reduce_window.mlir",
"remainder.mlir",
Expand All @@ -81,9 +82,7 @@ iree_check_single_backend_test_suite(
"while.mlir",
],
include = ["*.mlir"],
exclude = [
"pow.mlir", # TODO(#12678): Investigate this failing on riscv-32.
],
exclude = [],
),
compiler_flags = ["--iree-input-type=stablehlo"],
driver = "local-task",
Expand Down
1 change: 1 addition & 0 deletions tests/e2e/stablehlo_ops/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,7 @@ iree_check_single_backend_test_suite(
"multiply.mlir"
"negate.mlir"
"pad.mlir"
"pow.mlir"
"reduce.mlir"
"reduce_window.mlir"
"remainder.mlir"
Expand Down

0 comments on commit 97e6823

Please sign in to comment.