Update error message in meta__convert_weight_to_int4pack (pytorch#130707)

yanbing-j · pytorchmergebot · commit 93a03edcf995 · 2024-07-16T00:44:35.000Z
This PR is to fix error message in pytorch#129940. Pull Request resolved: pytorch#130707 Approved by: https://github.com/lezcano, https://github.com/malfet
diff --git a/torch/_meta_registrations.py b/torch/_meta_registrations.py
@@ -3256,7 +3256,7 @@ def meta__convert_weight_to_int4pack(w, inner_k_tiles):
     torch._check(w.dim() == 2, lambda: "w must be a 2D tensor")
     torch._check(
         w.dtype is torch.uint8,
-        lambda: f"expected w to be int32, got {w.dtype}",
+        lambda: f"expected w to be uint8, got {w.dtype}",
     )
     n = w.size(0)
     k = w.size(1)

Original file line number	Diff line number	Diff line change
`@@ -3256,7 +3256,7 @@ def meta__convert_weight_to_int4pack(w, inner_k_tiles):`
`3256`	`3256`	`torch._check(w.dim() == 2, lambda: "w must be a 2D tensor")`
`3257`	`3257`	`torch._check(`
`3258`	`3258`	`w.dtype is torch.uint8,`
`3259`		`- lambda: f"expected w to be int32, got {w.dtype}",`
	`3259`	`+ lambda: f"expected w to be uint8, got {w.dtype}",`
`3260`	`3260`	`)`
`3261`	`3261`	`n = w.size(0)`
`3262`	`3262`	`k = w.size(1)`