Skip to content

Commit

Permalink
Disable Flux transformer tests that hang (#748)
Browse files Browse the repository at this point in the history
Tests introduced by #741 would
actually hang during IREE compilation without one pending fix.
  • Loading branch information
sogartar authored Jan 3, 2025
1 parent 03a5918 commit d42cc29
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions sharktank/tests/models/flux/flux_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -170,6 +170,12 @@ def runCompareDevRandomSingleLayerIreeAgainstTorchEager(
raises=AssertionError,
reason="Accuracy is not good enough. The observed absolute error is 8976.53.",
)
@pytest.mark.skip(
reason=(
"Waiting on merging of fix for https://github.com/iree-org/iree/issues/19539. "
"Without it IREE compilation enters an infinite loop."
)
)
@with_flux_data
def testCompareDevRandomSingleLayerIreeBf16AgainstTorchEagerF32(self):
self.runCompareDevRandomSingleLayerIreeAgainstTorchEager(
Expand All @@ -180,6 +186,12 @@ def testCompareDevRandomSingleLayerIreeBf16AgainstTorchEagerF32(self):
raises=AssertionError,
reason="Accuracy is probably not good enough. The observed absolute error is 73.25.",
)
@pytest.mark.skip(
reason=(
"Waiting on merging of fix for https://github.com/iree-org/iree/issues/19539. "
"Without it IREE compilation enters an infinite loop."
)
)
@with_flux_data
def testCompareDevRandomSingleLayerIreeF32AgainstTorchEagerF32(self):
self.runCompareDevRandomSingleLayerIreeAgainstTorchEager(
Expand Down

0 comments on commit d42cc29

Please sign in to comment.