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 index a2426b93da19..e60836207114 100644 --- a/experimental/regression_suite/shark-test-suite-models/sdxl/test_unet.py +++ b/experimental/regression_suite/shark-test-suite-models/sdxl/test_unet.py @@ -292,13 +292,13 @@ def SDXL_PUNET_INT8_FP8_OUT( ] if os.path.isfile( - f"{iree_test_path_extension}/attention_and_matmul_spec_punet_{sku}.mlir" + f"{iree_test_path_extension}/attention_and_matmul_spec_punet_{sku}.mlir" ): 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 + # 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" )