1
- """
2
- @brief test log(time=3s)
3
- """
4
1
import os
5
2
import pprint
6
3
import unittest
@@ -383,25 +380,26 @@ def test_search_e5m2_pow(self):
383
380
)
384
381
385
382
def test_float32_to_fe4m3fn_inf (self ):
386
- v0 = numpy .float32 (numpy .nan )
383
+ mx =
384
+ v0 = numpy .float32 (448 )
387
385
v1 = numpy .float32 (numpy .inf )
388
386
a = search_float32_into_fe4m3 (v0 )
389
387
b = search_float32_into_fe4m3 (v1 )
390
388
self .assertEqual (a , b )
391
389
392
- v0 = numpy .float32 (numpy . nan )
390
+ v0 = numpy .float32 (448 )
393
391
v1 = numpy .float32 (numpy .inf )
394
392
a = float32_to_fe4m3 (v0 )
395
393
b = float32_to_fe4m3 (v1 )
396
394
self .assertEqual (a , b )
397
395
398
- v0 = numpy .float32 (- numpy . nan )
396
+ v0 = numpy .float32 (- 448 )
399
397
v1 = numpy .float32 (- numpy .inf )
400
398
a = search_float32_into_fe4m3 (v0 )
401
399
b = search_float32_into_fe4m3 (v1 )
402
400
self .assertEqual (a , b )
403
401
404
- v0 = numpy .float32 (- numpy . nan )
402
+ v0 = numpy .float32 (- 448 )
405
403
v1 = numpy .float32 (- numpy .inf )
406
404
a = float32_to_fe4m3 (v0 )
407
405
b = float32_to_fe4m3 (v1 )
@@ -619,25 +617,25 @@ def test_search_float32_into_fe5m2fnuz(self):
619
617
)
620
618
621
619
def test_float32_to_fe4m3fnuz_inf (self ):
622
- v0 = numpy .float32 (numpy . nan )
620
+ v0 = numpy .float32 (448 )
623
621
v1 = numpy .float32 (numpy .inf )
624
622
a = search_float32_into_fe4m3 (v0 , uz = True )
625
623
b = search_float32_into_fe4m3 (v1 , uz = True )
626
624
self .assertEqual (a , b )
627
625
628
- v0 = numpy .float32 (numpy . nan )
626
+ v0 = numpy .float32 (448 )
629
627
v1 = numpy .float32 (numpy .inf )
630
628
a = float32_to_fe4m3 (v0 , uz = True )
631
629
b = float32_to_fe4m3 (v1 , uz = True )
632
630
self .assertEqual (a , b )
633
631
634
- v0 = numpy .float32 (- numpy . nan )
632
+ v0 = numpy .float32 (- 448 )
635
633
v1 = numpy .float32 (- numpy .inf )
636
634
a = search_float32_into_fe4m3 (v0 , uz = True )
637
635
b = search_float32_into_fe4m3 (v1 , uz = True )
638
636
self .assertEqual (a , b )
639
637
640
- v0 = numpy .float32 (- numpy . nan )
638
+ v0 = numpy .float32 (- 448 )
641
639
v1 = numpy .float32 (- numpy .inf )
642
640
a = float32_to_fe4m3 (v0 , uz = True )
643
641
b = float32_to_fe4m3 (v1 , uz = True )
@@ -653,7 +651,7 @@ def test_float32_to_fe4m3fnuz_inf(self):
653
651
v1 = numpy .float32 (- numpy .inf )
654
652
a = search_float32_into_fe4m3 (v0 , uz = True )
655
653
b = search_float32_into_fe4m3 (v1 , uz = True )
656
- self .assertEqual (a , b )
654
+ self .assertNotEqual (a , b )
657
655
658
656
v0 = numpy .float32 (numpy .nan )
659
657
v1 = numpy .float32 (- numpy .nan )
@@ -665,16 +663,16 @@ def test_float32_to_fe4m3fnuz_inf(self):
665
663
v1 = numpy .float32 (- numpy .inf )
666
664
a = float32_to_fe4m3 (v0 , uz = True )
667
665
b = float32_to_fe4m3 (v1 , uz = True )
668
- self .assertEqual (a , b )
666
+ self .assertNotEqual (a , b )
669
667
670
668
def test_float32_to_fe5m2fnuz_inf (self ):
671
- v0 = numpy .float32 (numpy . nan )
669
+ v0 = numpy .float32 (65536 )
672
670
v1 = numpy .float32 (numpy .inf )
673
671
a = search_float32_into_fe5m2 (v0 , fn = True , uz = True )
674
672
b = search_float32_into_fe5m2 (v1 , fn = True , uz = True )
675
673
self .assertEqual (a , b )
676
674
677
- v0 = numpy .float32 (numpy . nan )
675
+ v0 = numpy .float32 (65536 )
678
676
v1 = numpy .float32 (numpy .inf )
679
677
a = float32_to_fe5m2 (v0 , fn = True , uz = True )
680
678
b = float32_to_fe5m2 (v1 , fn = True , uz = True )
0 commit comments