Fix build on some platforms

dsharletg · xnnpack-bot · commit 0a3103ab2930 · 2025-06-03T12:48:54.000-07:00
`_mm256_xor_epi32` is an AVX512 instruction (but `_mm256_xor_si256` is AVX2...)

PiperOrigin-RevId: 766781489
diff --git a/src/xnnpack/simd/f32-avx2.h b/src/xnnpack/simd/f32-avx2.h
@@ -69,7 +69,7 @@ static XNN_INLINE xnn_simd_f32_t xnn_cmpeq_f32(xnn_simd_f32_t a,
 
 static XNN_INLINE xnn_simd_f32_t xnn_cmpneq_f32(xnn_simd_f32_t a,
                                                 xnn_simd_f32_t b) {
-  return _mm256_castsi256_ps(_mm256_xor_epi32(
+  return _mm256_castsi256_ps(_mm256_xor_si256(
       _mm256_cmpeq_epi32(_mm256_castps_si256(a), _mm256_castps_si256(a)),
       _mm256_cmpeq_epi32(_mm256_castps_si256(a), _mm256_castps_si256(b))));
 }
diff --git a/src/xnnpack/simd/f32-fma3.h b/src/xnnpack/simd/f32-fma3.h
@@ -91,7 +91,7 @@ static XNN_INLINE xnn_simd_f32_t xnn_cmpeq_f32(xnn_simd_f32_t a,
 static XNN_INLINE xnn_simd_f32_t xnn_cmpneq_f32(xnn_simd_f32_t a,
                                                 xnn_simd_f32_t b) {
 #ifdef __AVX2__
-  return _mm256_castsi256_ps(_mm256_xor_epi32(
+  return _mm256_castsi256_ps(_mm256_xor_si256(
       _mm256_cmpeq_epi32(_mm256_castps_si256(a), _mm256_castps_si256(a)),
       _mm256_cmpeq_epi32(_mm256_castps_si256(a), _mm256_castps_si256(b))));
 #else

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ static XNN_INLINE xnn_simd_f32_t xnn_cmpeq_f32(xnn_simd_f32_t a,`
`69`	`69`
`70`	`70`	`static XNN_INLINE xnn_simd_f32_t xnn_cmpneq_f32(xnn_simd_f32_t a,`
`71`	`71`	`xnn_simd_f32_t b) {`
`72`		`- return _mm256_castsi256_ps(_mm256_xor_epi32(`
	`72`	`+ return _mm256_castsi256_ps(_mm256_xor_si256(`
`73`	`73`	`_mm256_cmpeq_epi32(_mm256_castps_si256(a), _mm256_castps_si256(a)),`
`74`	`74`	`_mm256_cmpeq_epi32(_mm256_castps_si256(a), _mm256_castps_si256(b))));`
`75`	`75`	`}`