diff --git a/onnxruntime/core/mlas/lib/mlasi.h b/onnxruntime/core/mlas/lib/mlasi.h index 648f195fd6..0f365159ac 100644 --- a/onnxruntime/core/mlas/lib/mlasi.h +++ b/onnxruntime/core/mlas/lib/mlasi.h @@ -1804,7 +1804,8 @@ MlasMaximumFloat32x4(MLAS_FLOAT32X4 Vector1, MLAS_FLOAT32X4 Vector2) #elif defined(MLAS_SSE2_INTRINSICS) return _mm_max_ps(Vector1, Vector2); #elif defined(MLAS_VSX_INTRINSICS) - return vec_max(Vector1, Vector2); + // Don't use vec_max to avoid undefined behavior if NAN + return vec_sel(Vector2, Vector1, vec_cmpgt(Vector1, Vector2)); #elif defined(MLAS_WASM_SIMD_INTRINSICS) return wasm_f32x4_max(Vector1, Vector2); #else @@ -1821,7 +1822,8 @@ MlasMinimumFloat32x4(MLAS_FLOAT32X4 Vector1, MLAS_FLOAT32X4 Vector2) #elif defined(MLAS_SSE2_INTRINSICS) return _mm_min_ps(Vector1, Vector2); #elif defined(MLAS_VSX_INTRINSICS) - return vec_min(Vector1, Vector2); + // Don't use vec_min to avoid undefined behavior if NAN + return vec_sel(Vector2, Vector1, vec_cmpgt(Vector2, Vector1)); #elif defined(MLAS_WASM_SIMD_INTRINSICS) return wasm_f32x4_min(Vector1, Vector2); #else