Skip to content

Commit

Permalink
Linting
Browse files Browse the repository at this point in the history
Signed-off-by: Ian <[email protected]>
  • Loading branch information
IanNod committed Jan 23, 2025
1 parent eb732d6 commit 12acf02
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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"
)
Expand Down

0 comments on commit 12acf02

Please sign in to comment.