Add missing checks for geometry shader stages in GrVkUtil.cpp

Bug: skia:
Change-Id: I30cb6d5ee5016536537c48346c58ef4cf19f16c5
Reviewed-on: https://skia-review.googlesource.com/21845
Reviewed-by: Greg Daniel <egdaniel@google.com>
Commit-Queue: Chris Dalton <csmartdalton@google.com>
diff --git a/src/gpu/vk/GrVkUtil.cpp b/src/gpu/vk/GrVkUtil.cpp
index c56a4fb..84406c8 100644
--- a/src/gpu/vk/GrVkUtil.cpp
+++ b/src/gpu/vk/GrVkUtil.cpp
@@ -231,6 +231,9 @@
     if (VK_SHADER_STAGE_VERTEX_BIT == stage) {
         return SkSL::Program::kVertex_Kind;
     }
+    if (VK_SHADER_STAGE_GEOMETRY_BIT == stage) {
+        return SkSL::Program::kGeometry_Kind;
+    }
     SkASSERT(VK_SHADER_STAGE_FRAGMENT_BIT == stage);
     return SkSL::Program::kFragment_Kind;
 }
@@ -239,6 +242,9 @@
     if (SkSL::Program::kVertex_Kind == kind) {
         return VK_SHADER_STAGE_VERTEX_BIT;
     }
+    if (SkSL::Program::kGeometry_Kind == kind) {
+        return VK_SHADER_STAGE_GEOMETRY_BIT;
+    }
     SkASSERT(SkSL::Program::kFragment_Kind == kind);
     return VK_SHADER_STAGE_FRAGMENT_BIT;
 }