File tree 1 file changed +12
-0
lines changed
sharktank/tests/models/flux
1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -170,6 +170,12 @@ def runCompareDevRandomSingleLayerIreeAgainstTorchEager(
170
170
raises = AssertionError ,
171
171
reason = "Accuracy is not good enough. The observed absolute error is 8976.53." ,
172
172
)
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
+ )
173
179
@with_flux_data
174
180
def testCompareDevRandomSingleLayerIreeBf16AgainstTorchEagerF32 (self ):
175
181
self .runCompareDevRandomSingleLayerIreeAgainstTorchEager (
@@ -180,6 +186,12 @@ def testCompareDevRandomSingleLayerIreeBf16AgainstTorchEagerF32(self):
180
186
raises = AssertionError ,
181
187
reason = "Accuracy is probably not good enough. The observed absolute error is 73.25." ,
182
188
)
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
+ )
183
195
@with_flux_data
184
196
def testCompareDevRandomSingleLayerIreeF32AgainstTorchEagerF32 (self ):
185
197
self .runCompareDevRandomSingleLayerIreeAgainstTorchEager (
You can’t perform that action at this time.
0 commit comments