update attention test threshold

microsoft · Mar 18, 2024 · 8d3b231 · 8d3b231
1 parent c6f66af
commit 8d3b231
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/onnxruntime/test/contrib_ops/attention_op_test.cc b/onnxruntime/test/contrib_ops/attention_op_test.cc
@@ -228,7 +228,7 @@ static void RunAttentionTest(
     }
 
     if (use_float16) {
-      tester.SetOutputTolerance(0.005f, 0.005f);
+      tester.SetOutputTolerance(0.005f);
     } else {
       tester.SetOutputTolerance(0.001f, 0.001f);
     }

diff --git a/onnxruntime/test/contrib_ops/packed_multihead_attention_op_test.cc b/onnxruntime/test/contrib_ops/packed_multihead_attention_op_test.cc
@@ -107,7 +107,7 @@ static void RunPackedMultiHeadAttentionTest(
       }
 
       tester.AddOutput<MLFloat16>("output", output_dims, ToFloat16(output_data));
-      tester.SetOutputTolerance(0.005f, 0.005f);
+      tester.SetOutputTolerance(0.005f);
     } else {
       if (is_packed_qkv) {
         tester.AddInput<float>("query", packed_qkv_dims, query_data);

diff --git a/orttraining/orttraining/test/training_ops/cuda/batch_norm_internal_test.cc b/orttraining/orttraining/test/training_ops/cuda/batch_norm_internal_test.cc
@@ -68,6 +68,7 @@ static void TestBatchNormInternal(bool test_double = false, bool T_is_half = fal
     test.AddOutput<double>("running_var", channel_dims, running_var_double);
     test.AddOutput<double>("saved_mean", channel_dims, saved_mean_double);
     test.AddOutput<double>("saved_inv_std", channel_dims, saved_inv_std_double);
+    test.SetOutputTolerance(0.0001f);
   } else {
     if (T_is_half) {
       std::vector<MLFloat16> X_half(X.size());