If no spec file for sku uses mi300 for punet due to numeric issues wi…

…thout Signed-off-by: Ian <[email protected]> Signed-off-by: ianNod <[email protected]>
iree-org · Jan 23, 2025 · cb5dcf0 · cb5dcf0
1 parent 70a152c
commit cb5dcf0
Showing 1 changed file with 5 additions and 0 deletions.
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
@@ -297,6 +297,11 @@ def SDXL_PUNET_INT8_FP8_OUT(
     INT8_PUNET_FLAGS.append(
         f"--iree-codegen-transform-dialect-library={iree_test_path_extension}/attention_and_matmul_spec_punet_{sku}.mlir"
     )
+else:
+    # TODO: Investigate numerics failure without using the MI300 punet attention spec 
+    INT8_PUNET_FLAGS.append(
+        f"--iree-codegen-transform-dialect-library={iree_test_path_extension}/attention_and_matmul_spec_punet_mi300.mlir"
+    )
 
 
 ROCM_UNET_PIPELINE_FP16_COMPILE_FLAGS = [