diff --git a/backends/qualcomm/builders/node_visitor.py b/backends/qualcomm/builders/node_visitor.py index f21bae7b4a8..7380f7a8191 100644 --- a/backends/qualcomm/builders/node_visitor.py +++ b/backends/qualcomm/builders/node_visitor.py @@ -69,6 +69,7 @@ torch.uint8: PyQnnManager.Qnn_DataType_t.QNN_DATATYPE_UINT_8, torch.uint16: PyQnnManager.Qnn_DataType_t.QNN_DATATYPE_UINT_16, torch.uint32: PyQnnManager.Qnn_DataType_t.QNN_DATATYPE_UINT_32, + bool: PyQnnManager.Qnn_DataType_t.QNN_DATATYPE_UINT_32, float: PyQnnManager.Qnn_DataType_t.QNN_DATATYPE_FLOAT_32, int: PyQnnManager.Qnn_DataType_t.QNN_DATATYPE_UINT_32, }