Allow pinning quantized tensors. (#10873)

This commit is contained in:
comfyanonymous
2025-11-24 23:48:20 -08:00
committed by GitHub
parent 25022e0b09
commit b6805429b9
2 changed files with 13 additions and 1 deletions

View File

@@ -228,6 +228,14 @@ class QuantizedTensor(torch.Tensor):
new_kwargs = dequant_arg(kwargs)
return func(*new_args, **new_kwargs)
def data_ptr(self):
return self._qdata.data_ptr()
def is_pinned(self):
return self._qdata.is_pinned()
def is_contiguous(self):
return self._qdata.is_contiguous()
# ==============================================================================
# Generic Utilities (Layout-Agnostic Operations)