GeeeekExplorer commited on
Commit
553034d
·
verified ·
1 Parent(s): 6976c7f

Update inference/kernel.py

Browse files
Files changed (1) hide show
  1. inference/kernel.py +1 -1
inference/kernel.py CHANGED
@@ -85,7 +85,7 @@ def act_quant_kernel(
85
  for i, j in T.Parallel(blk_m, group_size):
86
  y_local[i, j] = T.Cast(
87
  out_dtype,
88
- T.Cast(compute_dtype, T.Cast(out_dtype, T.clamp(
89
  x_local[i, j] / s_local[i], fp8_min, fp8_max
90
  ))) * s_local[i],
91
  )
 
85
  for i, j in T.Parallel(blk_m, group_size):
86
  y_local[i, j] = T.Cast(
87
  out_dtype,
88
+ T.Cast(compute_dtype, T.Cast(FP8, T.clamp(
89
  x_local[i, j] / s_local[i], fp8_min, fp8_max
90
  ))) * s_local[i],
91
  )