Fix InvSqrt causing nans

Auburn · Dec 22, 2023 · 197b578 · 197b578
1 parent 85f5543
commit 197b578
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/include/FastNoise/Generators/Modifiers.inl b/include/FastNoise/Generators/Modifiers.inl
@@ -280,7 +280,9 @@ class FastSIMD::DispatchClass<FastNoise::SquareRoot, SIMD> final : public virtua
     {
         float32v value = this->GetSourceValue( mSource, seed, pos... );
 
-        return FS::InvSqrt( FS::Max( FS::Abs( value ), float32v( FLT_MIN ) ) ) * value;
+        float32v invSqrt = FS::InvSqrt( FS::Abs( value ) );
+
+        return FS::Masked( invSqrt != float32v( INFINITY ), value * invSqrt );
     }
 };
 

diff --git a/include/FastNoise/Generators/Utils.inl b/include/FastNoise/Generators/Utils.inl
@@ -251,7 +251,9 @@ namespace FastNoise
                 float32v distSqr = dX * dX;
                 ((distSqr = FS::FMulAdd( d, d, distSqr )), ...);
 
-                return FS::InvSqrt( distSqr ) * distSqr;
+                float32v invSqrt = FS::InvSqrt( distSqr );
+
+                return FS::Masked( invSqrt != float32v( INFINITY ), distSqr * invSqrt );
             }
 
             case DistanceFunction::EuclideanSquared: