Skip to content

Commit

Permalink
Fix lint issue
Browse files Browse the repository at this point in the history
  • Loading branch information
Xia-Weiwen committed May 6, 2024
1 parent e1b60d3 commit 95c29a6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions bitsandbytes/backends/cpu_xpu_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,9 @@ def _maybe_torch_compile(func):
# Don't use torch.compile for now due to PyTorch issue https://github.com/pytorch/pytorch/issues/124382
def double_quant_impl(A, col_stats=None, row_stats=None, out_col=None, out_row=None, threshold=0.0):
"""
Find absolute max valus of each row/column of a tensor, and symmetrically quantize it to int8.
Find absolute max values of each row/column of a tensor, and symmetrically quantize it to int8.
If threshold > 0.0, only values <= threshold are counted. All outliers are zeroed out in
the original tensor and they are kept in COO format: (rows, cols, valus)
the original tensor and they are kept in COO format: (rows, cols, values)
If threshold == 0.0, there are no outliers.
Args:
A The tensor to be analyzed and quantized.
Expand Down

0 comments on commit 95c29a6

Please sign in to comment.