Skip to content

Commit 50c4d52

Browse files
committed
fix inf
1 parent 338686d commit 50c4d52

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

_unittests/ut_validation/test_f8.py

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,3 @@
1-
"""
2-
@brief test log(time=3s)
3-
"""
41
import os
52
import pprint
63
import unittest
@@ -383,25 +380,26 @@ def test_search_e5m2_pow(self):
383380
)
384381

385382
def test_float32_to_fe4m3fn_inf(self):
386-
v0 = numpy.float32(numpy.nan)
383+
mx =
384+
v0 = numpy.float32(448)
387385
v1 = numpy.float32(numpy.inf)
388386
a = search_float32_into_fe4m3(v0)
389387
b = search_float32_into_fe4m3(v1)
390388
self.assertEqual(a, b)
391389

392-
v0 = numpy.float32(numpy.nan)
390+
v0 = numpy.float32(448)
393391
v1 = numpy.float32(numpy.inf)
394392
a = float32_to_fe4m3(v0)
395393
b = float32_to_fe4m3(v1)
396394
self.assertEqual(a, b)
397395

398-
v0 = numpy.float32(-numpy.nan)
396+
v0 = numpy.float32(-448)
399397
v1 = numpy.float32(-numpy.inf)
400398
a = search_float32_into_fe4m3(v0)
401399
b = search_float32_into_fe4m3(v1)
402400
self.assertEqual(a, b)
403401

404-
v0 = numpy.float32(-numpy.nan)
402+
v0 = numpy.float32(-448)
405403
v1 = numpy.float32(-numpy.inf)
406404
a = float32_to_fe4m3(v0)
407405
b = float32_to_fe4m3(v1)
@@ -619,25 +617,25 @@ def test_search_float32_into_fe5m2fnuz(self):
619617
)
620618

621619
def test_float32_to_fe4m3fnuz_inf(self):
622-
v0 = numpy.float32(numpy.nan)
620+
v0 = numpy.float32(448)
623621
v1 = numpy.float32(numpy.inf)
624622
a = search_float32_into_fe4m3(v0, uz=True)
625623
b = search_float32_into_fe4m3(v1, uz=True)
626624
self.assertEqual(a, b)
627625

628-
v0 = numpy.float32(numpy.nan)
626+
v0 = numpy.float32(448)
629627
v1 = numpy.float32(numpy.inf)
630628
a = float32_to_fe4m3(v0, uz=True)
631629
b = float32_to_fe4m3(v1, uz=True)
632630
self.assertEqual(a, b)
633631

634-
v0 = numpy.float32(-numpy.nan)
632+
v0 = numpy.float32(-448)
635633
v1 = numpy.float32(-numpy.inf)
636634
a = search_float32_into_fe4m3(v0, uz=True)
637635
b = search_float32_into_fe4m3(v1, uz=True)
638636
self.assertEqual(a, b)
639637

640-
v0 = numpy.float32(-numpy.nan)
638+
v0 = numpy.float32(-448)
641639
v1 = numpy.float32(-numpy.inf)
642640
a = float32_to_fe4m3(v0, uz=True)
643641
b = float32_to_fe4m3(v1, uz=True)
@@ -653,7 +651,7 @@ def test_float32_to_fe4m3fnuz_inf(self):
653651
v1 = numpy.float32(-numpy.inf)
654652
a = search_float32_into_fe4m3(v0, uz=True)
655653
b = search_float32_into_fe4m3(v1, uz=True)
656-
self.assertEqual(a, b)
654+
self.assertNotEqual(a, b)
657655

658656
v0 = numpy.float32(numpy.nan)
659657
v1 = numpy.float32(-numpy.nan)
@@ -665,16 +663,16 @@ def test_float32_to_fe4m3fnuz_inf(self):
665663
v1 = numpy.float32(-numpy.inf)
666664
a = float32_to_fe4m3(v0, uz=True)
667665
b = float32_to_fe4m3(v1, uz=True)
668-
self.assertEqual(a, b)
666+
self.assertNotEqual(a, b)
669667

670668
def test_float32_to_fe5m2fnuz_inf(self):
671-
v0 = numpy.float32(numpy.nan)
669+
v0 = numpy.float32(65536)
672670
v1 = numpy.float32(numpy.inf)
673671
a = search_float32_into_fe5m2(v0, fn=True, uz=True)
674672
b = search_float32_into_fe5m2(v1, fn=True, uz=True)
675673
self.assertEqual(a, b)
676674

677-
v0 = numpy.float32(numpy.nan)
675+
v0 = numpy.float32(65536)
678676
v1 = numpy.float32(numpy.inf)
679677
a = float32_to_fe5m2(v0, fn=True, uz=True)
680678
b = float32_to_fe5m2(v1, fn=True, uz=True)

0 commit comments

Comments
 (0)