This commit is contained in:
qazal 2026-06-15 16:24:49 +08:00
commit 8941fec7ca

View file

@ -359,7 +359,7 @@ __global__ __launch_bounds__(512, 2) void hk_fp8_gemm(bf16 *C_ptr, fp8e4m3 *A_pt
#endif
#if SCALE_MODE & 4
float g_scale = (*g_scale_ptr + 1e-08f) * (1.0f / 448.0f);
scale *= *g_scale;
scale *= g_scale;
#endif
mul(cA, cA, scale);