add xfail for mi250 compile and change vmfb names

Signed-off-by: saienduri <[email protected]>
iree-org · Nov 9, 2024 · bc759b3 · bc759b3
1 parent 9843be9
commit bc759b3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/experimental/benchmarks/sdxl/benchmark_sdxl_rocm.py b/experimental/benchmarks/sdxl/benchmark_sdxl_rocm.py
@@ -122,7 +122,7 @@ def run_sdxl_punet_int8_fp16_rocm_benchmark(rocm_chip):
         "iree-benchmark-module",
         f"--device=hip",
         "--device_allocator=caching",
-        f"--module={punet_int8_fp16_dir_compile}/punet.rocm_{rocm_chip}.vmfb",
+        f"--module={punet_int8_fp16_dir_compile}/punet_fp16.rocm_{rocm_chip}.vmfb",
         f"--parameters=model={punet_int8_fp16_dir}/punet_weights.irpa",
         "--function=main",
         f"--input=1x4x128x128xf16",
@@ -292,7 +292,7 @@ def test_sdxl_rocm_benchmark(
         )
         logging.getLogger().info(compilation_line)
 
-        module_path = f"{punet_int8_fp16_dir_compile}/punet.rocm_{rocm_chip}.vmfb"
+        module_path = f"{punet_int8_fp16_dir_compile}/punet_fp16.rocm_{rocm_chip}.vmfb"
         punet_int8_fp16_binary_size = Path(module_path).stat().st_size
         compilation_line = (
             f"Punet F16 Binary Size: {punet_int8_fp16_binary_size} bytes"

diff --git a/experimental/regression_suite/shark-test-suite-models/sdxl/test_unet.py b/experimental/regression_suite/shark-test-suite-models/sdxl/test_unet.py
@@ -316,6 +316,13 @@ def test_run_unet_fp16_rocm(
 
 
 def test_compile_punet_int8_fp16_rocm(sdxl_punet_int8_fp16_mlir):
+    if rocm_chip == "gfx90a":
+        request.node.add_marker(
+            pytest.mark.xfail(
+                reason="Expected punet_int8_fp8 compilation on mi250 to fail",
+                strict=True,
+            )
+        )
     VmfbManager.sdxl_punet_int8_fp16_rocm_vmfb = iree_compile(
         sdxl_punet_int8_fp16_mlir,
         ROCM_COMPILE_FLAGS + INT8_PUNET_FLAGS,