Skip to content

Commit a90003f

Browse files
authored
ENH Make BufferDict repr accelerator agnostic (#2731)
Signed-off-by: Yao, Matrix <matrix.yao@intel.com>
1 parent 9b420cc commit a90003f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/peft/tuners/_buffer_dict.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,8 @@ def extra_repr(self):
148148
child_lines = []
149149
for k, p in self._buffers.items():
150150
size_str = "x".join(str(size) for size in p.size())
151-
device_str = "" if not p.is_cuda else f" (GPU {p.get_device()})"
151+
device_type = p.device.type
152+
device_str = "" if device_type == "cpu" else f" ({device_type.upper()} {p.get_device()})"
152153
parastr = f"Buffer containing: [{torch.typename(p)} of size {size_str}{device_str}]"
153154
child_lines.append(" (" + k + "): " + parastr)
154155
tmpstr = "\n".join(child_lines)

0 commit comments

Comments
 (0)