Skip to content

Commit 55675a2

Browse files
committed
Update on "[ET-VK] Deserialize VkGraph in ET-VK"
Add logic to deserialize a VkGraph blob back python object. This allows us to get a implement debugging / visualization directly on the vulkan-exported program. Still extra works need to be done: From the entire bundle, need to extract the specific vulkan delegate first. Differential Revision: [D66443780](https://our.internmc.facebook.com/intern/diff/D66443780/) [ghstack-poisoned]
2 parents 78c2bb3 + ec4b919 commit 55675a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/vulkan/test/test_serialization.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def _generate_random_const_tensors(self, num_tensors: int) -> List[torch.Tensor]
4646

4747
return tensors
4848

49-
def test_serialize_vulkan_binary(self):
49+
def test_serialize_vulkan_binary(self) -> None:
5050
vk_graph = VkGraph(
5151
version="0",
5252
chain=[],
@@ -104,7 +104,7 @@ def test_serialize_vulkan_binary(self):
104104
tensor_bytes = bytes(array)
105105
self.assertEqual(constant_data_bytes, tensor_bytes)
106106

107-
def test_serialize_deserialize_vkgraph(self):
107+
def test_serialize_deserialize_vkgraph(self) -> None:
108108
in_vk_graph = VkGraph(
109109
version="1",
110110
chain=[

0 commit comments

Comments
 (0)