diff --git a/backends/vulkan/partitioner/vulkan_partitioner.py b/backends/vulkan/partitioner/vulkan_partitioner.py index 5745d7c4cc..90d858f15c 100644 --- a/backends/vulkan/partitioner/vulkan_partitioner.py +++ b/backends/vulkan/partitioner/vulkan_partitioner.py @@ -50,6 +50,10 @@ def node_val_is_compatible(self, node_val: Any) -> bool: if len(node_val.shape) > 4: return False + # bool dtype not currently supported + if node_val.dtype == torch.bool: + return False + if isinstance(node_val, (list, tuple)): for item in node_val: if not self.node_val_is_compatible(item):