[ET-VK][ez] Fix 8-bit linear shaders extension requirement (#7607)

## Context 8-bit linear compute shaders only require the 8bit_storage extension for buffer tensors. Now that we are checking that the GPU supports required extensions, we need to be more careful about declaring that a extensions as required. Differential Revision: [D68035429](https://our.internmc.facebook.com/intern/diff/D68035429/) ghstack-source-id: 260953913 Pull Request resolved: #7600 Co-authored-by: Stephen Jia <[email protected]>
pytorch · Jan 10, 2025 · 73f50df · 73f50df
1 parent 9666ee8
commit 73f50df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backends/vulkan/runtime/graph/ops/glsl/q_8w_linear.glsl b/backends/vulkan/runtime/graph/ops/glsl/q_8w_linear.glsl
@@ -16,7 +16,8 @@
 ${define_active_storage_type(STORAGE)}
 
 ${define_required_extensions(DTYPE)}
-${define_required_extensions("int8")}
+$if STORAGE == "buffer":
+  ${define_required_extensions("int8")}
 
 #include "indexing_utils.h"
 

diff --git a/backends/vulkan/runtime/graph/ops/glsl/q_8w_linear_optimized.glsl b/backends/vulkan/runtime/graph/ops/glsl/q_8w_linear_optimized.glsl
@@ -16,7 +16,8 @@
 ${define_active_storage_type(STORAGE)}
 
 ${define_required_extensions(DTYPE)}
-${define_required_extensions("int8")}
+$if STORAGE == "buffer":
+  ${define_required_extensions("int8")}
 
 
 $if BATCH_MODE: