We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 457bdf1 commit 6c3f4a6Copy full SHA for 6c3f4a6
torchao/utils.py
@@ -573,7 +573,7 @@ class PlainAQTTensorImpl(...):
573
__torch_function__ = classmethod(_dispatch__torch_function__)
574
register_layout = classmethod(_register_layout)
575
get_tensor_impl_constructor = classmethod(_get_tensor_impl_constructor)
576
- _get_to_kwargs = _get_to_kwargs
+ _get_to_kwargs = classmethod(_get_to_kwargs)
577
578
def __tensor_flatten__(self):
579
raise NotImplementedError("Subclasses must implement __tensor_flatten__")
0 commit comments