mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
c
This commit is contained in:
parent
9db573a6d3
commit
8941fec7ca
1 changed files with 1 additions and 1 deletions
|
|
@ -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);
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue