Update ops.py

matthewdouglas · web-flow · commit 1542c188bd22 · 2025-06-05T10:05:33.000-04:00
diff --git a/bitsandbytes/backends/hpu/ops.py b/bitsandbytes/backends/hpu/ops.py
@@ -24,7 +24,6 @@ def _(
         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}")
 
     # Enable non uint8 dtype
     if A.dtype != torch.uint8:

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,6 @@ def _(`
`24`	`24`	`A.dtype in [torch.bfloat16, torch.uint8],`
`25`	`25`	`lambda: f"quant_storage supports uint8 or bfloat16, but got {A.dtype}",`
`26`	`26`	`)`
`27`		`- torch._check(A.dtype in [torch.bfloat16, torch.uint8], lambda: f"quant_storage supports uint8 or bfloat16, but got {A.dtype}")`
`28`	`27`
`29`	`28`	`# Enable non uint8 dtype`
`30`	`29`	`if A.dtype != torch.uint8:`