mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

compute

Browse source
This commit is contained in:
qazal 2026-06-15 16:23:37 +08:00
commit 9db573a6d3
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
extra/thunder/amd/gemm_fp8.cpp
View file

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