[ET-VK][ez] Handle zero-element tensors when building Vulkan graph (pytorch#12021)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: pytorch#11984 by
@SS-JIA
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/250/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/250/head
Merge bot PR base:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/249/orig
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/250/orig
@diff-train-skip-merge

cc @SS-JIA @manuelcandales @cbilgin

---------

Co-authored-by: Stephen Jia <[email protected]>

Author: pytorchbot

backends/vulkan/serialization/vulkan_graph_serialize.py

@@ -191,19 +191,23 @@ def serialize_constant_tensors(
 
     current_offset = len(raw_bytes)
     for tensor in const_tensors:
-        array_type = ctypes.c_char * tensor.untyped_storage().nbytes()
-        array = ctypes.cast(
-            tensor.untyped_storage().data_ptr(),
-            ctypes.POINTER(array_type),
-        ).contents
-
-        tensor_bytes = bytes(array)
-        # Pad the tensor bytes to the next 16 byte boundary
-        raw_bytes += tensor_bytes
-        raw_bytes += b"\x00" * padding_required(len(tensor_bytes))
-
-        vk_graph.constants.append(VkBytes(current_offset, len(tensor_bytes)))
-        current_offset += aligned_size(len(tensor_bytes))
+        if tensor.numel() == 0:
+            vk_graph.constants.append(VkBytes(current_offset, 0))
+            continue
+        else:
+            array_type = ctypes.c_char * tensor.untyped_storage().nbytes()
+            array = ctypes.cast(
+                tensor.untyped_storage().data_ptr(),
+                ctypes.POINTER(array_type),
+            ).contents
+
+            tensor_bytes = bytes(array)
+            # Pad the tensor bytes to the next 16 byte boundary
+            raw_bytes += tensor_bytes
+            raw_bytes += b"\x00" * padding_required(len(tensor_bytes))
+
+            vk_graph.constants.append(VkBytes(current_offset, len(tensor_bytes)))
+            current_offset += aligned_size(len(tensor_bytes))
 
 
 def serialize_custom_shaders(