Skip to content

Commit f02a5d6

Browse files
committed
fix: Fixed sawb_utils perCh tensor creation
Signed-off-by: Brandon Groth <[email protected]>
1 parent 33a899c commit f02a5d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fms_mo/quant_refactor/sawb_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def sawb_params_code(
114114
# conv W=[ch_o, ch_i, ki, ij], linear W=[ch_o, ch_i], reduce all dim but ch_out
115115
mu = torch.mean(input_tensor.abs(), dim=reduce_dim)
116116
std = torch.mean(input_tensor**2, dim=reduce_dim).sqrt()
117-
clip_val_vec = torch.tensor(coeff[1] * mu + coeff[0] * std)
117+
clip_val_vec = coeff[1] * mu + coeff[0] * std
118118
return None, clip_val_vec
119119

120120
# per-tensor

0 commit comments

Comments
 (0)