Update ops.py

matthewdouglas · web-flow · commit 3900187617f5 · 2025-06-05T09:46:37.000-04:00
Fix lint issue
diff --git a/bitsandbytes/backends/hpu/ops.py b/bitsandbytes/backends/hpu/ops.py
@@ -21,7 +21,8 @@ def _(
     torch._check_is_size(blocksize)
     torch._check(quant_type == "nf4", lambda: f"quant_type must be nf4, got {quant_type}")
     torch._check(
-        dtype in (torch.bfloat16, torch.float32), lambda: f"4bit dequantization only bf16/f32, but got {dtype}"
+        A.dtype in [torch.bfloat16, torch.uint8],
+        lambda: f"quant_storage supports uint8 or bfloat16, but got {A.dtype}",
     )
     torch._check(A.dtype in [torch.bfloat16, torch.uint8], lambda: f"quant_storage supports uint8 or bfloat16, but got {A.dtype}")
 

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,8 @@ def _(`
`21`	`21`	`torch._check_is_size(blocksize)`
`22`	`22`	`torch._check(quant_type == "nf4", lambda: f"quant_type must be nf4, got {quant_type}")`
`23`	`23`	`torch._check(`
`24`		`- dtype in (torch.bfloat16, torch.float32), lambda: f"4bit dequantization only bf16/f32, but got {dtype}"`
	`24`	`+ A.dtype in [torch.bfloat16, torch.uint8],`
	`25`	`+ lambda: f"quant_storage supports uint8 or bfloat16, but got {A.dtype}",`
`25`	`26`	`)`
`26`	`27`	`torch._check(A.dtype in [torch.bfloat16, torch.uint8], lambda: f"quant_storage supports uint8 or bfloat16, but got {A.dtype}")`
`27`	`28`