torch.bool support

Reviewed By: jaybean-dev, jfix71 Differential Revision: D48089238 fbshipit-source-id: a15492d96a03d149d36a7c682bfb7bb6ba5bcdf1
pytorch · Aug 7, 2023 · e208dd2 · e208dd2
1 parent 5ad7151
commit e208dd2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/Graph/FXIRUtils.cpp b/lib/Graph/FXIRUtils.cpp
@@ -48,6 +48,8 @@ const std::unordered_map<std::string, ElemKind> stringToElemKind = {
     {"acc.uint8fused", ElemKind::UInt8FusedQTy},
     // 4-bit fused quantize type
     {"acc.uint4fused", ElemKind::UInt4FusedQTy},
+    // bool type
+    {"torch.bool", ElemKind::BoolTy},
 };
 }