add more fpsqrt specialcases

diff --git a/src/ieee754/fpdiv/specialcases.py b/src/ieee754/fpdiv/specialcases.py
index 47e87abb6821292b1ad5923b3f5336476006df76..5c491638e059a7bb06ab0d08ea3d52d78737cdf0 100644 (file)
--- a/src/ieee754/fpdiv/specialcases.py
+++ b/src/ieee754/fpdiv/specialcases.py
@@ -114,6 +114,11 @@ class FPDIVSpecialCasesMod(Elaboratable):
                 m.d.comb += self.o.out_do_z.eq(1)
                 m.d.comb += self.o.z.inf(sabx)
 
+            # if a is NaN return NaN
+            with m.If(a1.is_nan):
+                m.d.comb += self.o.out_do_z.eq(1)
+                m.d.comb += self.o.z.nan(0)
+
             # if a is zero return zero
             with m.Elif(a1.is_zero):
                 m.d.comb += self.o.out_do_z.eq(1)