Skip to content

GitLab

Unverified Commit ea75e9f1 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Use `assertAlmostEqual` in `BloomEmbeddingTest.test_logits` (#19200) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 98af4f9b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/bloom/test_modeling_bloom.py
View file @ ea75e9f1
...@@ -771,8 +771,8 @@ class BloomEmbeddingTest(unittest.TestCase): ...@@ -771,8 +771,8 @@ class BloomEmbeddingTest(unittest.TestCase):
output_gpu_1, output_gpu_2 = output.split(125440, dim=-1) output_gpu_1, output_gpu_2 = output.split(125440, dim=-1)
if cuda_available: if cuda_available:
self.assertEqual(output_gpu_1.mean().item(), MEAN_LOGITS_GPU_1) self.assertAlmostEqual(output_gpu_1.mean().item(), MEAN_LOGITS_GPU_1, places=6)
self.assertEqual(output_gpu_2.mean().item(), MEAN_LOGITS_GPU_2) self.assertAlmostEqual(output_gpu_2.mean().item(), MEAN_LOGITS_GPU_2, places=6)
else: else:
self.assertAlmostEqual(output_gpu_1.mean().item(), MEAN_LOGITS_GPU_1, places=6) # 1e-06 precision!! self.assertAlmostEqual(output_gpu_1.mean().item(), MEAN_LOGITS_GPU_1, places=6) # 1e-06 precision!!
self.assertAlmostEqual(output_gpu_2.mean().item(), MEAN_LOGITS_GPU_2, places=6) self.assertAlmostEqual(output_gpu_2.mean().item(), MEAN_LOGITS_GPU_2, places=6)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment