From e292a138a3ad0f0f6b2cd7a4600424def7d32f89 Mon Sep 17 00:00:00 2001
From: Jorge Pineda <jorgep31415@meta.com>
Date: Mon, 7 Oct 2024 09:05:06 -0700
Subject: [PATCH] [5/n][ET-VK] Enable `uint8` dtype in shaders

TSIA

Differential Revision: [D63918660](https://our.internmc.facebook.com/intern/diff/D63918660/)

[ghstack-poisoned]
---
 backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.cpp b/backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.cpp
index 81d5c9e98af..469c2ed8280 100644
--- a/backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.cpp
+++ b/backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.cpp
@@ -47,6 +47,10 @@ void add_dtype_suffix(std::string& kernel_name, const vkapi::ScalarType dtype) {
     case vkapi::kQInt8:
       kernel_name += "_int8";
       break;
+    case vkapi::kByte:
+    case vkapi::kQUInt8:
+      kernel_name += "_uint8";
+      break;
     default:
       break;
   }