Skip to content

Commit 41141b7

Browse files
authored
Remove the iree compile xfail (#1251)
iree-org/iree#20491 has fix the f16 compiler issue So remove the iree compile xfail.
1 parent e8ae7bd commit 41141b7

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

sharktank/tests/models/llama/benchmark_amdgpu_test.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,6 @@ def setUp(self):
225225
]
226226

227227
@skipif_run_quick_llama_test
228-
@pytest.mark.xfail(
229-
reason="Iree Compile Error", strict=True, raises=IreeCompileException
230-
)
231228
def testBenchmark8B_f16_TP1_Non_Decomposed_Input_Len_128(self):
232229
output_file_name = self.dir_path_8b / "f16_torch_128_tp1"
233230
output_mlir = self.llama8b_f16_torch_sdpa_artifacts.create_file(
@@ -273,9 +270,6 @@ def testBenchmark8B_f16_TP1_Non_Decomposed_Input_Len_128(self):
273270
)
274271

275272
@skipif_run_quick_llama_test
276-
@pytest.mark.xfail(
277-
reason="Iree Compile Error", strict=True, raises=IreeCompileException
278-
)
279273
def testBenchmark8B_f16_TP1_Non_Decomposed_Input_Len_2048(self):
280274
output_file_name = self.dir_path_8b / "f16_torch_2048_tp1"
281275
output_mlir = self.llama8b_f16_torch_sdpa_artifacts.create_file(

0 commit comments

Comments
 (0)