diff --git a/backends/vulkan/serialization/vulkan_graph_builder.py b/backends/vulkan/serialization/vulkan_graph_builder.py index f94779aab9..4877bd6a86 100644 --- a/backends/vulkan/serialization/vulkan_graph_builder.py +++ b/backends/vulkan/serialization/vulkan_graph_builder.py @@ -248,6 +248,9 @@ def get_or_create_value_for(self, arg: _Argument): return self.create_scalar_list_value(arg) elif isinstance(arg, list) and isinstance(arg[0], Node): return self.create_value_list_value(arg) + elif isinstance(arg, torch.fx.immutable_collections.immutable_list): + # pyre-ignore[6] + return self.create_value_list_value(arg) elif isinstance(arg, str): return self.create_string_value(arg) else: