Skip to content

Commit d42cc29

Browse files
authored
Disable Flux transformer tests that hang (#748)
Tests introduced by #741 would actually hang during IREE compilation without one pending fix.
1 parent 03a5918 commit d42cc29

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

sharktank/tests/models/flux/flux_test.py

+12
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,12 @@ def runCompareDevRandomSingleLayerIreeAgainstTorchEager(
170170
raises=AssertionError,
171171
reason="Accuracy is not good enough. The observed absolute error is 8976.53.",
172172
)
173+
@pytest.mark.skip(
174+
reason=(
175+
"Waiting on merging of fix for https://github.com/iree-org/iree/issues/19539. "
176+
"Without it IREE compilation enters an infinite loop."
177+
)
178+
)
173179
@with_flux_data
174180
def testCompareDevRandomSingleLayerIreeBf16AgainstTorchEagerF32(self):
175181
self.runCompareDevRandomSingleLayerIreeAgainstTorchEager(
@@ -180,6 +186,12 @@ def testCompareDevRandomSingleLayerIreeBf16AgainstTorchEagerF32(self):
180186
raises=AssertionError,
181187
reason="Accuracy is probably not good enough. The observed absolute error is 73.25.",
182188
)
189+
@pytest.mark.skip(
190+
reason=(
191+
"Waiting on merging of fix for https://github.com/iree-org/iree/issues/19539. "
192+
"Without it IREE compilation enters an infinite loop."
193+
)
194+
)
183195
@with_flux_data
184196
def testCompareDevRandomSingleLayerIreeF32AgainstTorchEagerF32(self):
185197
self.runCompareDevRandomSingleLayerIreeAgainstTorchEager(

0 commit comments

Comments
 (0)