Skip to content

Commit e9e6671

Browse files
authored
Add more logs to TensorCoreTiledAQTLayout (#520)
Summary: att, for easier debugging Test Plan: . Reviewers: Subscribers: Tasks: Tags:
1 parent ec95afd commit e9e6671

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/dtypes/affine_quantized_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -512,7 +512,7 @@ def to(self, *args, **kwargs):
512512
kwargs = self._get_to_kwargs(*args, **kwargs)
513513
device = kwargs["device"]
514514
if device != "cuda" or (isinstance(device, torch.device) and device.type != "cuda"):
515-
raise ValueError(f"TensorCoreTiledAQTLayout is only available for cuda device")
515+
raise ValueError(f"TensorCoreTiledAQTLayout is only available for cuda device, can't convert to {device}")
516516
return self.__class__(
517517
self.packed_weight.to(device),
518518
self.scale_and_zero.to(device),

0 commit comments

Comments
 (0)