s

nihui · nihui · commit 845c730eb277 · 2026-03-25T11:50:53.000Z
diff --git a/src/layer/x86/gemm_bf16s.h b/src/layer/x86/gemm_bf16s.h
@@ -2515,7 +2515,7 @@ static void gemm_transB_packed_tile_bf16s(const Mat& AT_tile, const Mat& BT_tile
             for (; kk < max_kk; kk++)
             {
                 __m512 _pA0 = bfloat2float_avx512(_mm256_loadu_si256((const __m256i*)pA));
-                __m512 _pB0 = bfloat2float_avx512(_mm256_set1_ps(((const float*)pB)[0]));
+                __m512 _pB0 = bfloat2float_avx512(_mm256_castps_si256(_mm256_set1_ps(((const float*)pB)[0])));
                 __m512 _pB1 = _mm512_permute_ps(_pB0, _MM_SHUFFLE(0, 3, 2, 1));
 
                 _sum0 = _mm512_fmadd_ps(_pA0, _pB0, _sum0);

Original file line number	Diff line number	Diff line change
`@@ -2515,7 +2515,7 @@ static void gemm_transB_packed_tile_bf16s(const Mat& AT_tile, const Mat& BT_tile`
`2515`	`2515`	`for (; kk < max_kk; kk++)`
`2516`	`2516`	`{`
`2517`	`2517`	`__m512 _pA0 = bfloat2float_avx512(_mm256_loadu_si256((const __m256i*)pA));`
`2518`		`- __m512 _pB0 = bfloat2float_avx512(_mm256_set1_ps(((const float*)pB)[0]));`
	`2518`	`+ __m512 _pB0 = bfloat2float_avx512(_mm256_castps_si256(_mm256_set1_ps(((const float*)pB)[0])));`
`2519`	`2519`	`__m512 _pB1 = _mm512_permute_ps(_pB0, _MM_SHUFFLE(0, 3, 2, 1));`
`2520`	`2520`
`2521`	`2521`	`_sum0 = _mm512_fmadd_ps(_pA0, _pB0, _sum0);`