[Vulkan] Remove GLSL Code Gen (pytorch#91912)

@bypass-github-export-checks

GLSL Code Gen is not used, so this diff removes
- GLSL parts of ShaderSource
- Anything enclosed by USE_VULKAN_SHADERC_RUNTIME, as well as the flag itself
- gen_vulkan_glsl script

Plus some additional refactoring

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

**NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D41358861/)!
Pull Request resolved: pytorch#91912
Approved by: https://github.com/mcr229

Author: salilsdesai

CMakeLists.txt

@@ -266,7 +266,6 @@ option(USE_SOURCE_DEBUG_ON_MOBILE "Enable " ON)
 option(USE_LITE_INTERPRETER_PROFILER "Enable " ON)
 option(USE_VULKAN_FP16_INFERENCE "Vulkan - Use fp16 inference" OFF)
 option(USE_VULKAN_RELAXED_PRECISION "Vulkan - Use relaxed precision math in the kernels (mediump)" OFF)
-option(USE_VULKAN_SHADERC_RUNTIME "Vulkan - Use runtime shader compilation as opposed to build-time (needs libshaderc)" OFF)
 # option USE_XNNPACK: try to enable xnnpack by default.
 option(USE_XNNPACK "Use XNNPACK" ON)
 option(USE_ZMQ "Use ZMQ" OFF)
@@ -746,9 +745,6 @@ if(USE_VULKAN)
     string(APPEND CMAKE_CXX_FLAGS " -DUSE_VULKAN_RELAXED_PRECISION")
   endif()
 
-  if(USE_VULKAN_SHADERC_RUNTIME)
-    string(APPEND CMAKE_CXX_FLAGS " -DUSE_VULKAN_SHADERC_RUNTIME")
-  endif()
 endif()
 
 if(BUILD_LITE_INTERPRETER)

aten/src/ATen/gen_vulkan_glsl.py

@@ -1,9 +1,5 @@
 #include <ATen/native/vulkan/api/Shader.h>
 
-#ifdef USE_VULKAN_SHADERC_RUNTIME
-#include <shaderc/shaderc.hpp>
-#endif /* USE_VULKAN_SHADERC_RUNTIME */
-
 namespace at {
 namespace native {
 namespace vulkan {
@@ -14,38 +10,19 @@ namespace api {
 //
 
 ShaderInfo::ShaderInfo()
-    : type(ShaderInfo::Type::SPIRV),
-      src_code{
-          .spirv =
-              {
-                  nullptr,
-                  0u,
-              },
+    : src_code{
+          nullptr,
+          0u,
       } {}
 
-ShaderInfo::ShaderInfo(std::string name, const char* const glsl_src)
-    : type(ShaderInfo::Type::GLSL),
-      src_code{
-          .glsl =
-              {
-                  glsl_src,
-                  0u,
-              },
-      },
-      kernel_name{std::move(name)} {}
-
 ShaderInfo::ShaderInfo(
     std::string name,
     const uint32_t* const spirv_bin,
     const uint32_t size,
     const std::vector<VkDescriptorType>& layout)
-    : type(Type::SPIRV),
-      src_code{
-          .spirv =
-              {
-                  spirv_bin,
-                  size,
-              },
+    : src_code{
+          spirv_bin,
+          size,
       },
       kernel_name{std::move(name)},
       kernel_layout{layout} {}
@@ -58,13 +35,9 @@ ShaderInfo::ShaderInfo(
     const std::vector<uint32_t>& tile_size,
     const StorageType bias_storage_type,
     const StorageType weight_storage_type)
-    : type(Type::SPIRV),
-      src_code{
-          .spirv =
-              {
-                  spirv_bin,
-                  size,
-              },
+    : src_code{
+          spirv_bin,
+          size,
       },
       kernel_name{std::move(name)},
       kernel_layout{layout},
@@ -77,17 +50,9 @@ ShaderInfo::ShaderInfo(
 }
 
 bool operator==(const ShaderInfo& _1, const ShaderInfo& _2) {
-  if (_1.type != _2.type) {
-    return false;
-  }
-
-  if (_1.type == ShaderInfo::Type::SPIRV) {
-    return (
-        _1.src_code.spirv.bin == _2.src_code.spirv.bin &&
-        _1.src_code.spirv.size == _2.src_code.spirv.size);
-  } else {
-    return (_1.src_code.glsl.src == _2.src_code.glsl.src);
-  }
+  return (
+      _1.src_code.bin == _2.src_code.bin &&
+      _1.src_code.size == _2.src_code.size);
 }
 
 //
@@ -153,8 +118,8 @@ void swap(ShaderLayout& lhs, ShaderLayout& rhs) noexcept {
 
 ShaderModule::ShaderModule(const VkDevice device, const ShaderInfo& source)
     : device_(device), handle_{VK_NULL_HANDLE} {
-  const uint32_t* code = source.src_code.spirv.bin;
-  uint32_t size = source.src_code.spirv.size;
+  const uint32_t* code = source.src_code.bin;
+  uint32_t size = source.src_code.size;
 
   const VkShaderModuleCreateInfo shader_module_create_info{
       VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO, // sType

aten/src/ATen/native/vulkan/api/Shader.cpp

@@ -45,17 +45,9 @@ class ShaderLayout final {
 };
 
 struct ShaderInfo final {
-  enum class Type { GLSL, SPIRV } type;
-
-  union {
-    struct {
-      const char* src; // Null-terminated
-      uint32_t unused; // padding
-    } glsl;
-    struct {
-      const uint32_t* bin;
-      uint32_t size;
-    } spirv;
+  struct {
+    const uint32_t* bin;
+    uint32_t size;
   } src_code;
 
   std::string kernel_name{""};
@@ -171,8 +163,7 @@ class ShaderCache final {
 
   struct Hasher {
     inline size_t operator()(const ShaderInfo& source) const {
-      return c10::get_hash(
-          source.type, source.src_code.spirv.bin, source.src_code.spirv.size);
+      return c10::get_hash(source.src_code.bin, source.src_code.size);
     }
   };
 

aten/src/ATen/native/vulkan/api/Shader.h

@@ -6,18 +6,9 @@
 #include <ATen/core/Tensor.h>
 #include <ATen/native/vulkan/api/api.h>
 #include <ATen/native/vulkan/ops/Convert.h>
-
-#define CONCAT_LITERALS(a, b) #a #b
-#ifdef USE_VULKAN_SHADERC_RUNTIME
-#include <ATen/native/vulkan/glsl.h>
-#define VK_KERNEL(name)                          \
-  ::at::native::vulkan::api::ShaderInfo {        \
-    CONCAT_LITERALS(vulkan., name), name##_glsl, \
-  }
-#else
 #include <ATen/native/vulkan/spv.h>
+
 #define VK_KERNEL(name) ::at::native::vulkan::name##_spv
-#endif /* USE_VULKAN_SHADERC_RUNTIME */
 
 namespace at {
 namespace native {

aten/src/ATen/native/vulkan/ops/Common.h

@@ -175,7 +175,6 @@ function(caffe2_print_configuration_summary)
   if(${USE_VULKAN})
     message(STATUS "    USE_VULKAN_FP16_INFERENCE    : ${USE_VULKAN_FP16_INFERENCE}")
     message(STATUS "    USE_VULKAN_RELAXED_PRECISION : ${USE_VULKAN_RELAXED_PRECISION}")
-    message(STATUS "    USE_VULKAN_SHADERC_RUNTIME   : ${USE_VULKAN_SHADERC_RUNTIME}")
   endif()
   message(STATUS "  USE_PROF              : ${USE_PROF}")
   message(STATUS "  USE_QNNPACK           : ${USE_QNNPACK}")