@@ -197,7 +197,6 @@ def testBenchmark8B_f16_Decomposed(self):
197
197
)
198
198
199
199
@skipif_run_quick_llama_test
200
- @pytest .mark .xfail (reason = "Compile Error" , strict = True , raises = IreeCompileException )
201
200
def testBenchmark8B_f16_Non_Decomposed_Prefill (self ):
202
201
output_file_name = self .dir_path_8b / "f16_torch_prefill"
203
202
output_mlir = self .llama8b_f16_torch_sdpa_artifacts .create_file (
@@ -780,7 +779,9 @@ def testBenchmark405B_f16_TP8_Decomposed(self):
780
779
cwd = self .repo_root ,
781
780
)
782
781
783
- @pytest .mark .xfail (reason = "Compile Error" , strict = True , raises = IreeCompileException )
782
+ @pytest .mark .xfail (
783
+ reason = "Benchmarking Error" , strict = True , raises = IreeBenchmarkException
784
+ )
784
785
def testBenchmark405B_f16_TP8_Non_Decomposed (self ):
785
786
output_file_name = self .dir_path_405b / "f16_torch"
786
787
output_mlir = self .llama405b_f16_torch_sdpa_artifacts .create_file (
@@ -828,7 +829,9 @@ def testBenchmark405B_f16_TP8_Non_Decomposed(self):
828
829
cwd = self .repo_root ,
829
830
)
830
831
831
- @pytest .mark .xfail (reason = "Compile Error" , strict = True , raises = IreeCompileException )
832
+ @pytest .mark .xfail (
833
+ reason = "KeyError in theta.py" , strict = True , raises = ExportMlirException
834
+ )
832
835
def testBenchmark405B_fp8_TP8_Decomposed (self ):
833
836
output_file_name = self .dir_path_405b / "fp8_decomposed"
834
837
output_mlir = self .llama405b_fp8_decomposed_artifacts .create_file (
@@ -874,7 +877,9 @@ def testBenchmark405B_fp8_TP8_Decomposed(self):
874
877
cwd = self .repo_root ,
875
878
)
876
879
877
- @pytest .mark .xfail (reason = "Compile Error" , strict = True , raises = IreeCompileException )
880
+ @pytest .mark .xfail (
881
+ reason = "KeyError in theta.py" , strict = True , raises = ExportMlirException
882
+ )
878
883
def testBenchmark405B_fp8_TP8_Non_Decomposed (self ):
879
884
output_file_name = self .dir_path_405b / "fp8_torch"
880
885
output_mlir = self .llama405b_fp8_torch_sdpa_artifacts .create_file (
0 commit comments