Skip to content

Commit

Permalink
add back hz fusion
Browse files Browse the repository at this point in the history
Signed-off-by: saienduri <[email protected]>
  • Loading branch information
saienduri committed Nov 9, 2024
1 parent bc759b3 commit ea2be7f
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from pathlib import Path

vmfb_dir = os.getenv("TEST_OUTPUT_ARTIFACTS", default=Path.cwd())
rocm_chip = os.getenv("ROCM_CHIP", default="gfx90a")
rocm_chip = os.getenv("ROCM_CHIP", default="gfx942")
iree_test_path_extension = os.getenv("IREE_TEST_PATH_EXTENSION", default=Path.cwd())

###############################################################################
Expand Down Expand Up @@ -195,6 +195,7 @@ def SDXL_PUNET_INT8_FP8_OUT(
f"--iree-hip-target={rocm_chip}",
"--iree-opt-const-eval=false",
"--iree-global-opt-propagate-transposes=true",
"--iree-dispatch-creation-enable-fuse-horizontal-contractions=true",
"--iree-dispatch-creation-enable-aggressive-fusion=true",
"--iree-opt-aggressively-propagate-transposes=true",
"--iree-opt-outer-dim-concat=true",
Expand Down Expand Up @@ -315,7 +316,7 @@ def test_run_unet_fp16_rocm(
)


def test_compile_punet_int8_fp16_rocm(sdxl_punet_int8_fp16_mlir):
def test_compile_punet_int8_fp16_rocm(request, sdxl_punet_int8_fp16_mlir):
if rocm_chip == "gfx90a":
request.node.add_marker(
pytest.mark.xfail(
Expand Down

0 comments on commit ea2be7f

Please sign in to comment.