diff --git a/backends/intel_hpu/tests/config.py b/backends/intel_hpu/tests/config.py index 7f8329ca83..f005f60882 100644 --- a/backends/intel_hpu/tests/config.py +++ b/backends/intel_hpu/tests/config.py @@ -21,7 +21,5 @@ skip_case_lst = [ "test_cast.py", "test_fused_block_attention.py", - "test_fused_rms_mlp.py", - "test_fused_rms_qkv_rope.py", "test_index_copy.py", ] diff --git a/backends/intel_hpu/tests/unittests/test_fused_rms_mlp.py b/backends/intel_hpu/tests/unittests/test_fused_rms_mlp.py index 6366bed267..2172299578 100644 --- a/backends/intel_hpu/tests/unittests/test_fused_rms_mlp.py +++ b/backends/intel_hpu/tests/unittests/test_fused_rms_mlp.py @@ -115,8 +115,8 @@ def NP_Fused_RMS_MLP_OP( ) return np_mlp_out_ref - def check_result(self, np_result, fused_result): - np.testing.assert_allclose(np_result, fused_result) + def check_result(self, np_result, fused_result, rtol=1e-3): + np.testing.assert_allclose(np_result, fused_result, rtol) def test_fused_mlp(self): ( diff --git a/backends/intel_hpu/tests/unittests/test_fused_rms_qkv_rope.py b/backends/intel_hpu/tests/unittests/test_fused_rms_qkv_rope.py index 9b928cb2c7..824d921fe7 100644 --- a/backends/intel_hpu/tests/unittests/test_fused_rms_qkv_rope.py +++ b/backends/intel_hpu/tests/unittests/test_fused_rms_qkv_rope.py @@ -146,7 +146,7 @@ def check_result( op_result_value_states, ): if self.dtype == "float32": - rtol = 1e-6 + rtol = 1e-2 atol = 1e1 elif self.dtype == "float16": rtol = 1e-3