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 769684a commit 2283294Copy full SHA for 2283294
backends/qualcomm/quantizer/annotators.py
@@ -68,7 +68,7 @@ def _is_float_tensor(node: Node):
68
or not isinstance(node.meta["val"], FakeTensor)
69
):
70
return False
71
- return node.meta["val"].dtype == torch.float32
+ return node.meta["val"].dtype in (torch.bfloat16, torch.float32)
72
73
74
def _mark_nodes_as_annotated(nodes: List[Node]):
0 commit comments