From 54580faea972f559de6b68305a0958c77b421d64 Mon Sep 17 00:00:00 2001 From: Stephen Jia Date: Tue, 12 Nov 2024 08:01:22 -0800 Subject: [PATCH] [ET-VK] Replace `uint16_t` with `int` for swiftshader As title. This is a follow up from https://github.com/pytorch/executorch/pull/6675. Replace `uint16_t` with `int` as well as 16 bit vectorized types. Differential Revision: [D65824204](https://our.internmc.facebook.com/intern/diff/D65824204/) ghstack-source-id: 253112193 Pull Request resolved: https://github.com/pytorch/executorch/pull/6781 --- backends/vulkan/runtime/gen_vulkan_spv.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backends/vulkan/runtime/gen_vulkan_spv.py b/backends/vulkan/runtime/gen_vulkan_spv.py index 39d023e7658..2ac7bb98608 100644 --- a/backends/vulkan/runtime/gen_vulkan_spv.py +++ b/backends/vulkan/runtime/gen_vulkan_spv.py @@ -721,6 +721,7 @@ def maybe_replace_u16vecn(self, input_text: str) -> str: return input_text input_text = input_text.replace("u16vec", "ivec") + input_text = input_text.replace("uint16_t", "int") return input_text def generateSPV(self, output_dir: str) -> Dict[str, str]: