[ET-VK][ez] Fix 8-bit linear shaders extension requirement

SS-JIA · web-flow · commit ce24066cafc5 · 2025-01-10T14:43:53.000-08:00
Differential Revision: D68035429 Pull Request resolved: #7600
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:
