lint

hidet-org · Dec 21, 2024 · 8cba3a5 · 8cba3a5
1 parent b8b7ae5
commit 8cba3a5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/python/hidet/graph/ops/quant/symmetric.py b/python/hidet/graph/ops/quant/symmetric.py
@@ -27,11 +27,13 @@ def __init__(self, w: TensorNode, quant_type: DataType, dims: Union[int, List[in
         if not isinstance(dims, (list, tuple)):
             dims = [dims]
 
-        # bf16 can't hold int16.max_value. Should convert to f32 first. 
-        # For another types pair is similar. 
+        # bf16 can't hold int16.max_value. Should convert to f32 first.
+        # For another types pair is similar.
         # For cases when float type can hold quant_type.max_value we can skip this step but leave it for simplicity
-        wm = compute(name='abs', shape=w.shape, 
-            fcompute=lambda *indices: if_then_else(w[indices] >= 0, cast(w[indices], f32), cast(-w[indices], f32))
+        wm = compute(
+            name='abs',
+            shape=w.shape,
+            fcompute=lambda *indices: if_then_else(w[indices] >= 0, cast(w[indices], f32), cast(-w[indices], f32)),
         )
 
         scale = cops.reduce(wm, dims, keep_dim=False, reduce_type='max')